Update app.py

app.py

@@ -7,25 +7,22 @@ from peft import PeftModel
 # --- Configuration ---
 BASE_MODEL_ID = "CohereForAI/aya-101"
 
-# Map the dropdown options to your 3 Hugging Face Model IDs
 MODEL_MAP = {
     "English to Angika": "snjev310/aya-101-english-angika",
     "Hindi to Angika": "snjev310/aya-101-hindi-angika",
     "Angika to English": "snjev310/aya-101-angika-english"
 }
 
-# Load Tokenizer globally (it's small and stays in CPU RAM)
 tokenizer = AutoTokenizer.from_pretrained(BASE_MODEL_ID)
 
-@spaces.GPU(duration=180) # 3 minutes to allow for 13B model loading + inference
+@spaces.GPU(duration=180) 
 def translate(text, model_choice):
     if not text.strip():
         return "Please enter text to translate."
     
     adapter_id = MODEL_MAP[model_choice]
     
-    # 1. Load Base Model in bfloat16 (Standard for Aya-101)
-    # Pro ZeroGPU has ~70GB VRAM, so we don't need 4-bit quantization
+    # Load Base Model
     base_model = AutoModelForSeq2SeqLM.from_pretrained(
         BASE_MODEL_ID,
         torch_dtype=torch.bfloat16,
@@ -33,16 +30,14 @@ def translate(text, model_choice):
         device_map="auto"
     )
     
-    # 2. Load the specific PEFT Adapter
+    # Load the specific Adapter
     model = PeftModel.from_pretrained(base_model, adapter_id)
     model.eval()
     
-    # 3. Prepare Input
-    # Using a prompt format helps the model understand the task
-    prompt = f"{model_choice}: {text}"
+    # Prepare Input - Using the direction as a prefix
+    prompt = f"translate {model_choice}: {text}"
     inputs = tokenizer(prompt, return_tensors="pt").to(model.device)
     
-    # 4. Generate
     with torch.no_grad():
         outputs = model.generate(
             **inputs, 
@@ -54,7 +49,7 @@ def translate(text, model_choice):
     
     result = tokenizer.decode(outputs[0], skip_special_tokens=True)
     
-    # 5. Cleanup (CRITICAL for ZeroGPU to release resources)
+    # Cleanup to free GPU for next user
     del model
     del base_model
     torch.cuda.empty_cache()
@@ -63,8 +58,8 @@ def translate(text, model_choice):
 
 # --- Gradio UI ---
 with gr.Blocks(theme=gr.themes.Soft()) as demo:
-    gr.Markdown("# 🗣️ Angika Multi-Translator")
-    gr.Markdown("Powered by **Aya-101** and **ZeroGPU**. Select your translation direction below.")
+    gr.Markdown("# 🗣️ Angika Machine Translation")
+    gr.Markdown("Select your translation direction below. This uses a 13B parameter model.")
     
     with gr.Row():
         with gr.Column():
@@ -73,25 +68,13 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
                 value="English to Angika", 
                 label="Select Translation Mode"
             )
-            input_text = gr.Textbox(
-                label="Input Text", 
-                placeholder="Type here...", 
-                lines=5
-            )
+            input_text = gr.Textbox(label="Input Text", placeholder="Type here...", lines=5)
             submit_btn = gr.Button("Translate", variant="primary")
             
         with gr.Column():
-            output_text = gr.Textbox(
-                label="Translated Text", 
-                lines=5, 
-                interactive=False
-            )
+            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=[
@@ -101,4 +84,19 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
         inputs=[input_text, model_dropdown]
     )
 
+    gr.Markdown("---")
+    with gr.Row():
+        with gr.Column():
+            gr.Markdown("### 🛠️ Help us improve!")
+            gr.Markdown("Share the correct version so we can retrain the model.")
+            feedback_btn = gr.Button("Submit Correct Translation", variant="secondary")
+            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.")
+            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"); }')
+
 demo.launch()