Update app.py

app.py

@@ -66,41 +66,54 @@ class SafeGeocoder:
             return None
 
 # Function to just load the model
-# Function that truly tests if the model is loaded
+
 def load_model():
     try:
-        # Send a minimal request just to trigger model loading
+        # Use a simple but complete example to test actual extraction
+        test_template = '{"test_location": ""}'
+        test_text = "Heute sind wir in Berlin."
+        
+        prompt = f"<|input|>\n### Template:\n{test_template}\n### Text:\n{test_text}\n\n<|output|>"
+        
+        # Send a complete request with a simple test case
         payload = {
-            "inputs": "<|input|>\n### Template:\n{\"test\": \"\"}\n### Text:\ntest\n\n<|output|>",
+            "inputs": prompt,
             "parameters": {
-                "max_new_tokens": 10,
+                "max_new_tokens": 50,
                 "do_sample": False
             }
         }
         
+        # Update status to show we're starting
+        gr.Info("Modell wird geladen... Dies kann einen Moment dauern.")
+        
+        # Make the actual request
         response = requests.post(API_URL, headers=headers, json=payload)
         
+        # Check loading status
         if response.status_code == 503:
             response_json = response.json()
             if "error" in response_json and "loading" in response_json["error"]:
                 estimated_time = response_json.get("estimated_time", "unknown")
                 return f"⏳ Modell lädt... (ca. {int(float(estimated_time)) if isinstance(estimated_time, (int, float, str)) else 'unbekannt'} Sekunden)"
         
-        # The key difference: Actually check if we got valid output
+        # Verify we got a proper response
         if response.status_code == 200:
-            # Make sure we got a valid response not just HTTP 200
             result = response.json()
+            
+            # Check for a properly formatted extraction result
             if isinstance(result, list) and len(result) > 0:
-                # Try to verify we got real model output
                 result_text = result[0].get("generated_text", "")
-                if len(result_text) > 10:  # Some basic validation
-                    return "✅ Modell erfolgreich geladen! Sie können jetzt mit der Extraktion beginnen."
+                
+                # Look for evidence of a completed extraction
+                if "<|output|>" in result_text and "Berlin" in result_text:
+                    return "✅ Modell erfolgreich geladen und getestet! Sie können jetzt mit der Extraktion beginnen."
         
-        # If we get here, something is still wrong
-        return "⚠️ Modell scheint nicht vollständig geladen zu sein. Bitte versuchen Sie es erneut."
+        # If we get here, the model response wasn't complete
+        return "⚠️ Modell-Test nicht erfolgreich. Bitte versuchen Sie es erneut in einigen Sekunden."
             
     except Exception as e:
-        return f"❌ Fehler: {str(e)}"
+        return f"❌ Fehler beim Laden des Modells: {str(e)}"
 
 def extract_info(template, text):
     try: