Update app.py

app.py

@@ -8,7 +8,7 @@ from peft import PeftModel
 # tokenizer = AutoTokenizer.from_pretrained(model_path, use_fast=False)
 # model = AutoModelForSeq2SeqLM.from_pretrained(model_path)
 
-model_path = "Sabbir772/BnT5Sy"
+model_path = "Sabbir772/banglat5-finetuned-bn-Ch"
 tokenizer = AutoTokenizer.from_pretrained(model_path, use_fast=False)
 model = AutoModelForSeq2SeqLM.from_pretrained(model_path)
 model.eval()
@@ -19,8 +19,8 @@ model.to(device)
 def translate(text, source_lang):
     if source_lang == "Bangla":
         prefix = "<BN>"
-    elif source_lang == "Sylheti":
-        prefix = "<SY>"
+    elif source_lang == "Chatgaiya":
+        prefix = "<CH>"
     else:
         return "Invalid language selected."
 
@@ -37,17 +37,17 @@ def translate(text, source_lang):
         )
     
     return tokenizer.decode(outputs[0], skip_special_tokens=True)
-
+    
 # Gradio interface
 iface = gr.Interface(
     fn=translate,
     inputs=[
         gr.Textbox(label="Input Text"),
-        gr.Radio(["Bangla", "Sylheti"], label="Source Language")
+        gr.Radio(["Bangla", "Chatgaiya"], label="Source Language")
     ],
     outputs=gr.Textbox(label="Translated Text"),
-    title="Bangla ↔ Sylheti Dialect Translator (Fine-tuned T5)",
-    description="Translate between Bangla and Sylheti using a LoRA-finetuned Flan-T5 model."
+    title="Bangla ↔ Chatgaiya Dialect Translator (Fine-tuned T5)",
+    description="Translate between Bangla and Chatgaiya using a LoRA-finetuned  model."
 )
 
 iface.launch()