Update app.py

app.py

@@ -85,11 +85,30 @@ class SafeGeocoder:
 
 # Corrected model loading function based on official usage example
 
+@spaces.GPU
 def extract_info(template, text):
     global tokenizer, model
     
+    # Load tokenizer if not loaded yet
     if tokenizer is None:
-        return "❌ Tokenizer nicht geladen", "Bitte zuerst auf 'Modell laden' klicken"
+        print("Tokenizer not loaded yet, loading now...")
+        try:
+            try:
+                from modelscope import AutoTokenizer as MSAutoTokenizer
+                tokenizer = MSAutoTokenizer.from_pretrained(MODEL_NAME, trust_remote_code=True)
+                print("Loaded tokenizer using modelscope AutoTokenizer")
+            except:
+                # Fall back to regular tokenizer
+                tokenizer = AutoTokenizer.from_pretrained(
+                    MODEL_NAME, 
+                    trust_remote_code=True,
+                    revision="main"
+                )
+                print("Loaded tokenizer using standard AutoTokenizer")
+        except Exception as e:
+            trace = traceback.format_exc()
+            print(f"Error loading tokenizer: {e}\n{trace}")
+            return "❌ Fehler beim Laden des Tokenizers", f"{str(e)}"
     
     try:
         # Load model if not loaded yet