Update app.py

app.py

@@ -7,7 +7,7 @@ model_name = "jbochi/madlad400-3b-mt"
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = AutoModelForSeq2SeqLM.from_pretrained(
     model_name,
-    torch_dtype=torch.float16,  # Use float16 to reduce memory usage
+    torch_dtype=torch.float16,
     device_map="auto"
 )
 
@@ -53,14 +53,14 @@ def translate_text(text, source_lang, target_lang):
     except Exception as e:
         return f"Error during translation: {str(e)}"
 
-# Create the Gradio interface
+# Gradio interface
 with gr.Blocks(title="English-Persian Translator") as demo:
     gr.Markdown(
         """
         # 🌍 English-Persian Translator
-        **Powered by MADLAD-400-3B Model**
+        **Author: Saman Zeitouinan**
         
-        Translate text between English and Persian using the state-of-the-art MADLAD-400 model.
+        Translate text between English and Persian.
         """
     )
     
@@ -122,9 +122,8 @@ with gr.Blocks(title="English-Persian Translator") as demo:
     )
 
 if __name__ == "__main__":
-    # Launch the app
     demo.launch(
-        server_name="0.0.0.0",  # Allow external access
-        share=False,  # Set to True to get a public URL
+        server_name="0.0.0.0",
+        share=False,
         debug=True
     )