Update app.py

app.py

@@ -1,8 +1,15 @@
+import torch
+from functools import partial
+original_torch_load = torch.load
+torch.load = partial(original_torch_load, weights_only=False)
 
+# Now rest of imports
 import gradio as gr
 import unicodedata
 from fairseq.models.transformer import TransformerModel
 
+
+
 # ── Load model once at startup ──────────────────────────────────────────────
 model = TransformerModel.from_pretrained(
     model_name_or_path=".",