Update app.py

app.py

@@ -3,8 +3,28 @@ import os
 import base64
 import threading
 
+login(token=os.environ["HF_TOKEN"])
+repo_id = os.environ["REPO_ID"]
+
+# Torch optimizations
+import torch
+torch.set_num_threads(1)
+if torch.cuda.is_available():
+    torch.backends.cudnn.benchmark = True
+
+# Download generation logic from private repo
+try:
+    generate_file = hf_hub_download(
+        repo_id=repo_id,
+        filename="gen1.py",
+        token=os.environ["HF_TOKEN"]
+    )
+    os.system(f"cp {generate_file} ./gen1.py")
+except Exception as e:
+    print(f"Error downloading files: {e}")
+
 # Import the generation wrapper
-from gen import setup_translation, translate_text as gen_translate
+from gen1 import setup_translation, translate_text as gen_translate
 
 # Logo and favicon setup
 LOGO_PATH = "static/logo.png"