Update app.py

app.py

@@ -13,6 +13,7 @@ MODEL_MAP = {
     "Angika to English": "snjev310/aya-101-angika-english"
 }
 
+# Load Tokenizer globally
 tokenizer = AutoTokenizer.from_pretrained(BASE_MODEL_ID)
 
 @spaces.GPU(duration=180) 
@@ -34,7 +35,7 @@ def translate(text, model_choice):
     model = PeftModel.from_pretrained(base_model, adapter_id)
     model.eval()
     
-    # Prepare Input - Using the direction as a prefix
+    # Prepare Input
     prompt = f"translate {model_choice}: {text}"
     inputs = tokenizer(prompt, return_tensors="pt").to(model.device)
     
@@ -49,7 +50,7 @@ def translate(text, model_choice):
     
     result = tokenizer.decode(outputs[0], skip_special_tokens=True)
     
-    # Cleanup to free GPU for next user
+    # Cleanup for ZeroGPU
     del model
     del base_model
     torch.cuda.empty_cache()
@@ -57,9 +58,10 @@ def translate(text, model_choice):
     return result
 
 # --- Gradio UI ---
-with gr.Blocks(theme=gr.themes.Soft()) as demo:
+# Theme removed from Blocks constructor for Gradio 6.0
+with gr.Blocks() as demo:
     gr.Markdown("# 🗣️ Angika Machine Translation")
-    gr.Markdown("Select your translation direction below. This uses a 13B parameter model.")
+    gr.Markdown("Select your translation direction below. This uses the Aya-101 13B model.")
     
     with gr.Row():
         with gr.Column():
@@ -74,7 +76,11 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
         with gr.Column():
             output_text = gr.Textbox(label="Translated Text", lines=5, interactive=False)
 
-    submit_btn.click(fn=translate, inputs=[input_text, model_dropdown], outputs=output_text)
+    submit_btn.click(
+        fn=translate, 
+        inputs=[input_text, model_dropdown], 
+        outputs=output_text
+    )
     
     gr.Examples(
         examples=[
@@ -88,15 +94,23 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
     with gr.Row():
         with gr.Column():
             gr.Markdown("### 🛠️ Help us improve!")
-            gr.Markdown("Share the correct version so we can retrain the model.")
+            gr.Markdown("If the translation isn't perfect, please share the correct version with us.")
             feedback_btn = gr.Button("Submit Correct Translation", variant="secondary")
-            feedback_btn.click(fn=None, _js='() => { window.open("https://forms.gle/FXspX7DxXHh5En6c7", "_blank"); }')
+            # Changed '_js' to 'js' for Gradio 6.0 compatibility
+            feedback_btn.click(
+                fn=None, 
+                js='() => { window.open("https://forms.gle/FXspX7DxXHh5En6c7", "_blank"); }'
+            )
 
         with gr.Column():
             gr.Markdown("### ⚡ Support our computation")
-            gr.Markdown("Your support helps us keep this service free and cover hosting costs.")
+            gr.Markdown("Your support helps us cover the hosting costs for this 13B model.")
             support_btn = gr.Button("Support Computation Costs ☕", variant="primary")
-            # ADD YOUR RAZORPAY/INSTAMOJO/KO-FI LINK HERE:
-            # support_btn.click(fn=None, _js='() => { window.open("https://ko-fi.com/YOUR_USERNAME", "_blank"); }')
+            # Replace the link below with your Razorpay/Ko-fi link
+            support_btn.click(
+                fn=None, 
+                js='() => { window.open("https://ko-fi.com/YOUR_USERNAME", "_blank"); }'
+            )
 
-demo.launch()
+# Theme added to launch method for Gradio 6.0
+demo.launch(theme=gr.themes.Soft())