Update app.py

app.py

@@ -8,7 +8,7 @@ app = FastAPI()
 # This function will be able to generate text
 # given an input.
 pipe = pipeline("text2text-generation", 
-model="google/flan-t5-small")
+model="mat27/medmnistPrueba")
  
 # Define a function to handle the GET request at `/generate`
 # The generate() function is defined as a FastAPI route that takes a 
@@ -16,11 +16,7 @@ model="google/flan-t5-small")
 # containing the generated text under the key "output"
 @app.get("/generate")
 def generate(text: str):
-    """
-    Using the text2text-generation pipeline from `transformers`, generate text
-    from the given input text. The model used is `google/flan-t5-small`, which
-    can be found [here](<https://huggingface.co/google/flan-t5-small>).
-    """
+
     # Use the pipeline to generate text from the given input text
     output = pipe(text)