Update app.py

app.py

@@ -2,6 +2,15 @@ import gradio as gr
 import shutil
 import os
 from MangaTranslator import MangaTranslator
+import torch
+
+_original_jit_load = torch.jit.load
+
+def _safe_jit_load(f, map_location=None, _extra_files=None):
+    # Force CPU mapping regardless of what the library asks for
+    return _original_jit_load(f, map_location=torch.device('cpu'), _extra_files=_extra_files)
+
+torch.jit.load = _safe_jit_load
 
 # 1. Initialize the Translator (runs once on startup)
 print("⏳ Initializing models... (This takes 30s)")