Update app.py

app.py

@@ -1,47 +1,49 @@
 import gradio as gr
 from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
-import torch
 
-# Load fine-tuned Assamese ↔ English model
+# Your model on Hugging Face
 MODEL_NAME = "Sazid2/assamese-english-translator"
+
+# Load model and tokenizer
 tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
 model = AutoModelForSeq2SeqLM.from_pretrained(MODEL_NAME)
 
-# Translation function
-def translate_text(text):
+def translate(text):
+    """Translate Assamese text → English"""
     if not text.strip():
-        return "⚠️ Please enter some text."
-    inputs = tokenizer(text, return_tensors="pt", truncation=True, padding=True)
-    with torch.no_grad():
-        outputs = model.generate(**inputs, max_length=128)
-    translation = tokenizer.decode(outputs[0], skip_special_tokens=True)
-    return translation
-
-# ---------------------------
-# Safe Model Info Section
-# ---------------------------
-model_info = (
-    "### 🧠 Model Info\n"
-    "- **Model Name:** [Sazid2/assamese-english-translator](https://huggingface.co/Sazid2/assamese-english-translator)\n"
-    "- **Base Model:** Helsinki-NLP/opus-mt-mul-en\n"
-    "- **Language Pair:** Assamese <-> English\n"
-    "- **Fine-tuned By:** Abu Sazid Ahmed\n"
-    "- **Training Epochs:** 8\n"
-    "- **BLEU Score:** 🏆 38.02\n"
-    "- **Framework:** 🤗 Transformers (PyTorch)\n"
-    "- **Dataset:** Custom Assamese-English parallel corpus (~20k sentences)\n"
-)
+        return "⚠️ Please enter Assamese text."
+    
+    inputs = tokenizer(text, return_tensors="pt", padding=True, truncation=True)
+    outputs = model.generate(**inputs, max_length=128)
+    translated = tokenizer.decode(outputs[0], skip_special_tokens=True)
+    return translated
+
+title = "🌐 Assamese → English Translator"
+description = """
+### Fine-tuned Neural Machine Translation  
+This model translates **Assamese sentences to English**.  
+It is based on the **Helsinki-NLP/opus-mt-mul-en** model and fine-tuned on Assamese-English parallel data.  
+
+- 🔤 **Source language:** Assamese  
+- 🌍 **Target language:** English  
+- 🧠 **BLEU Score:** 38.02  
+- 🧩 **Framework:** Hugging Face Transformers  
+"""
 
-# ---------------------------
+examples = [
+    ["মই কামলৈ গৈছো।"],
+    ["তুমি ক'ত আছা?"],
+    ["সেইখন অত্যন্ত ধুনীয়া।"]
+]
 
-# Gradio UI
+# Build Gradio interface
 demo = gr.Interface(
-    fn=translate_text,
-    inputs=gr.Textbox(label="Enter Assamese or English text", placeholder="Type here..."),
-    outputs=gr.Textbox(label="Translation"),
-    title="🇮🇳 Assamese ↔ English Translator",
-    description="Fine-tuned model for Assamese ↔ English translation using Helsinki-NLP/opus-mt-mul-en base.",
-    theme="soft"
+    fn=translate,
+    inputs=gr.Textbox(label="Enter Assamese text"),
+    outputs=gr.Textbox(label="English Translation"),
+    title=title,
+    description=description,
+    examples=examples,
 )
 
 if __name__ == "__main__":