Update app.py

app.py

@@ -1,13 +1,9 @@
 import gradio as gr
 from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
 
-# Load the saved model
-model_path = "https://huggingface.co/spaces/jjuarez/translation/raw/main/files/"
-model = AutoModelForSeq2SeqLM.from_pretrained(model_path)
 
-# Load the tokenizer
-model_checkpoint = "domenicrosati/opus-mt-es-en-scielo"
-tokenizer = AutoTokenizer.from_pretrained(model_checkpoint)
+tokenizer = AutoTokenizer.from_pretrained("jjuarez/translation_model")
+model = AutoModelForSeq2SeqLM.from_pretrained("jjuarez/translation_model")
 
 # Define the translation function
 def translate(text):