Update app.py

app.py

@@ -37,7 +37,7 @@ def translate_msa_to_syrian(text):
     
     try:
         input_ids = msa_to_syrian_tokenizer(text, return_tensors="pt").input_ids.to(device)
-        outputs = msa_to_syrian_model.generate(input_ids, max_length=512, num_beams=5, early_stopping=True)
+        outputs = msa_to_syrian_model.generate(input_ids, max_length=128, num_beams=5, early_stopping=True)
         translated_text = msa_to_syrian_tokenizer.decode(outputs[0], skip_special_tokens=True)
         return translated_text
     except Exception as e:
@@ -51,7 +51,7 @@ def translate_syrian_to_msa(text):
     
     try:
         input_ids = syrian_to_msa_tokenizer(text, return_tensors="pt").input_ids.to(device)
-        outputs = syrian_to_msa_model.generate(input_ids, max_length=512, num_beams=5, early_stopping=True)
+        outputs = syrian_to_msa_model.generate(input_ids, max_length=128, num_beams=5, early_stopping=True)
         translated_text = syrian_to_msa_tokenizer.decode(outputs[0], skip_special_tokens=True)
         return translated_text
     except Exception as e:
@@ -67,7 +67,7 @@ def bidirectional_translate(text, direction):
         return "Please select a translation direction"
 
 # Create Gradio interface
-with gr.Blocks(title="SHAMI-MT: Bidirectional Arabic Translation") as demo:
+with gr.Blocks(title="SHAMI-MT: Bidirectional Syria Arabic Dialect MT Framework") as demo:
     
     gr.HTML("""
     <div style="text-align: center; margin-bottom: 2rem;">