Added Gradio translation app

app.py

@@ -12,10 +12,10 @@ def load_model():
     global tokenizer, model
     try:
         tokenizer = AutoTokenizer.from_pretrained(checkpoint_path)
-        model = AutoModelForSeq2SeqLM.from_pretrained(checkpoint_path)
+        model = AutoModelForSeq2SeqLM.from_pretrained(checkpoint_path, torch_dtype=torch.float16 if torch.cuda.is_available() else torch.float32)
         model.to(device)
         model.eval()
-        print(f"Model loaded successfully from Hugging Face on {device}!")
+        print(f"Model loaded successfully on {device}!")
     except Exception as e:
         print(f"Error loading model: {e}")
 
@@ -165,5 +165,4 @@ with gr.Blocks(css=css) as demo:
     )
 
 if __name__ == "__main__":
-    # Updated to use 127.0.0.1:5000 instead of 0.0.0.0:5000
-    demo.launch(share=True, server_name="127.0.0.1", server_port=5000)
+    demo.launch(server_name="0.0.0.0", server_port=7860)

requirements.txt

@@ -2,3 +2,4 @@ gradio
 torch
 transformers
 huggingface_hub
+sentencepiece