Update app.py

app.py

@@ -3,12 +3,21 @@ import json
 import difflib
 from flair.models import SequenceTagger
 from flair.data import Sentence
+import torch
 
 app = FastAPI()
 
+# Workaround för PyTorch 2.6 weights_only issue
+original_load = torch.load
+def patched_load(*args, **kwargs):
+    kwargs["weights_only"] = False
+    return original_load(*args, **kwargs)
+
 # Ladda Flair flerspråkig NER-modell
 try:
+    torch.load = patched_load
     tagger = SequenceTagger.load("flair/ner-multi")
+    torch.load = original_load  # Återställ original torch.load
 except Exception as e:
     print(f"Error loading model: {str(e)}")
     raise e