Update app.py

app.py

@@ -87,23 +87,34 @@ class SafeGeocoder:
 def load_model():
     global tokenizer, model
     try:
-        if model is None:
-            # Load tokenizer exactly as shown in the usage example
+        # Try to import Qwen2 components from modelscope
+        try:
+            from modelscope import AutoTokenizer as MSAutoTokenizer
+            tokenizer = MSAutoTokenizer.from_pretrained(MODEL_NAME, trust_remote_code=True)
+            model = AutoModelForCausalLM.from_pretrained(
+                MODEL_NAME,
+                torch_dtype=TORCH_DTYPE,
+                trust_remote_code=True
+            ).to(DEVICE).eval()
+            print("Loaded model using modelscope AutoTokenizer")
+        except:
+            # If modelscope approach fails, try with specific revision
             tokenizer = AutoTokenizer.from_pretrained(
                 MODEL_NAME, 
-                trust_remote_code=True
+                trust_remote_code=True,
+                revision="main"  # Try specifying a revision
             )
             
-            # Load model exactly as shown in the usage example
             model = AutoModelForCausalLM.from_pretrained(
                 MODEL_NAME,
                 torch_dtype=TORCH_DTYPE,
-                trust_remote_code=True
+                trust_remote_code=True,
+                revision="main"  # Try specifying a revision
             ).to(DEVICE).eval()
             
-            print(f"✅ Loaded {MODEL_NAME} on {DEVICE}")
-            
-            # Test the model
+        print(f"✅ Loaded {MODEL_NAME} on {DEVICE}")
+        
+        # Test the model
             test_text = "Test in Berlin."
             test_template = '{"test_location": ""}'
             test_template_formatted = json.dumps(json.loads(test_template), indent=4)
@@ -128,13 +139,13 @@ def load_model():
                 return "✅ Modell erfolgreich geladen und getestet!"
             
         return "⚠️ Modell-Test nicht erfolgreich. Bitte versuchen Sie es erneut."
-        
+
     except Exception as e:
         import traceback
         trace = traceback.format_exc()
         print(f"Error loading model: {e}\n{trace}")
         return f"❌ Fehler beim Laden des Modells: {str(e)}"
-
+        
 @spaces.GPU
 def extract_info(template, text):
     global tokenizer, model