Refactor translation endpoint

app.py

@@ -5,20 +5,17 @@ from transformers import T5Tokenizer, T5ForConditionalGeneration
 # Create a new FastAPI app instance
 app = FastAPI(docs_url="/", redoc_url="/new_redoc")
 
-# Initialize the text generation pipeline
-# This function will be able to generate text
-# given an input.
-# pipe = pipeline("translation", model="t5-small")
-
+# Create a translation pipeline
 tokenizer = T5Tokenizer.from_pretrained("google/flan-t5-small")
 model = T5ForConditionalGeneration.from_pretrained("google/flan-t5-small")
 
 
+# Create a POST endpoint
 @app.get("/generate")
 def generate(text: str):
-    """Translate a text input to another language"""
+    """Traduce del castellano al inglés"""
 
-    input_text = "translate English to German: " + text
+    input_text = "translate Spanish to English: " + text
     input_ids = tokenizer(input_text, return_tensors="pt").input_ids
 
     outputs = model.generate(input_ids)