Update app.py

app.py

@@ -15,8 +15,8 @@ import json
 embedder = None
 
 AVAILABLE_MODELS = {
-    "Qwen Original": "Qwen/Qwen3-Embedding-0.6B",
-    "Arabic Fine-tuned": "Omartificial-Intelligence-Space/Semantic-Ar-Qwen-Embed-0.6B"
+    "Qwen3-Embedding-0.6B": "Qwen/Qwen3-Embedding-0.6B",
+    "Semantic-Ar-Qwen-Embed-0.6B": "Omartificial-Intelligence-Space/Semantic-Ar-Qwen-Embed-0.6B"
 }
 
 class QwenEmbedder:
@@ -52,9 +52,9 @@ def initialize_embedder(embedding_dim=768):
     device = "cuda" if torch.cuda.is_available() else "cpu"
     print(f"Initializing embedder on device: {device}")
     
-    # Create and move model to device
+    # Create model with specified dimension
     model = QwenEmbedder(embedding_dim=embedding_dim)
-    return model.to_device(device)
+    return model
 
 @spaces.GPU(duration=120)
 def process_with_embedder(fn_name, *args):