Update app.py

app.py

@@ -4,22 +4,30 @@ import sys
 import base64
 import threading
 from huggingface_hub import login, hf_hub_download
-import torch
 
 login(token=os.environ["HF_TOKEN"])
 repo_id = os.environ["REPO_ID"]
 
+import torch
 torch.set_num_threads(1)
 if torch.cuda.is_available():
     torch.backends.cudnn.benchmark = True
 
 try:
-    generate_file = hf_hub_download(
-        repo_id=repo_id,
-        filename="generate.py",
-        token=os.environ["HF_TOKEN"]
-    )
-    os.system(f"cp {generate_file} ./generate.py")
+    files_to_download = ["generate.py", "model.py", "hfmod_sft.py"]
+    
+    for filename in files_to_download:
+        try:
+            file_path = hf_hub_download(
+                repo_id=repo_id,
+                filename=filename,
+                token=os.environ["HF_TOKEN"]
+            )
+            os.system(f"cp {file_path} ./{filename}")
+            print(f"Downloaded: {filename}")
+        except Exception as e:
+            print(f"Warning: Could not download {filename}: {e}")
+            
 except Exception as e:
     print(f"Error downloading files: {e}")
 
@@ -44,14 +52,9 @@ def init_translation_model():
         print("Failed to load model")
 
 def translate_text(source_text, source_lang, target_lang, temperature, top_k, repetition_penalty, max_tokens):
-    return chat_interface(
-        source_text, source_lang, target_lang, 
-        temperature, top_k, repetition_penalty, max_tokens
-    )
+    return chat_interface(source_text, source_lang, target_lang, temperature, top_k, repetition_penalty, max_tokens)
 
 languages = ["English", "Hindi", "Bengali", "Tamil", "Telugu", "Kannada", "Panjabi"]
-css_path = "static/style.css"
-custom_css = open(css_path, encoding="utf-8").read() if os.path.isfile(css_path) else ""
 
 theme_lock_css = """
 .gradio-container .theme-toggle,
@@ -70,7 +73,7 @@ body, .gradio-container {
 }
 """
 
-combined_css = custom_css + theme_lock_css
+combined_css = theme_lock_css
 
 locked_theme = gr.themes.Monochrome(primary_hue="blue", secondary_hue="slate", neutral_hue="slate").set(
     background_fill_primary="#0a1628",
@@ -83,9 +86,11 @@ locked_theme = gr.themes.Monochrome(primary_hue="blue", secondary_hue="slate", n
     body_text_color="#e6eef8"
 )
 
+# Create Gradio interface
 with gr.Blocks(
     title="Madhuram Translation - MaruthLabs",
-    css=combined_css, theme=locked_theme,
+    css=combined_css,
+    theme=locked_theme,
     js=f"""
     function() {{
         {f'document.head.insertAdjacentHTML("beforeend", `{FAVICON_HTML}`);' if FAVICON_HTML else ''}
@@ -169,23 +174,9 @@ with gr.Blocks(
     def clear_all():
         return "", ""
 
-    swap_btn.click(
-        fn=swap_languages, 
-        inputs=[source_lang, target_lang], 
-        outputs=[source_lang, target_lang]
-    )
-    
-    clear_btn.click(
-        fn=clear_all, 
-        outputs=[source_text, translated_text]
-    )
-    
-    translate_btn.click(
-        fn=translate_text, 
-        inputs=[source_text, source_lang, target_lang, temperature, top_k, repetition_penalty, max_tokens], 
-        outputs=[translated_text]
-    )
-    
+    swap_btn.click(fn=swap_languages, inputs=[source_lang, target_lang], outputs=[source_lang, target_lang])
+    clear_btn.click(fn=clear_all, outputs=[source_text, translated_text])
+    translate_btn.click(fn=translate_text, inputs=[source_text, source_lang, target_lang, temperature, top_k, repetition_penalty, max_tokens], outputs=[translated_text])
     demo.load(fn=init_translation_model)
 
 if __name__ == "__main__":