Upload app.py

app.py

@@ -6,7 +6,7 @@ import time
 import torch
 
 device = "cuda" if torch.cuda.is_available() else "cpu"
-generate_story_pipe = pipeline("text-generation", model="google/flan-t5-base")
+generate_story_pipe = pipeline("text2text-generation", model="google/flan-t5-base")
 generate_story_pipe = generate_story_pipe.to(device)
 
 # Para medir el rendimiento de los métodos, voy a crear este decorador, que simplemente imprime en nuestra terminal el tiempo de ejecucion de los metodos que tengan los modelos y los usen, de esta manera podremos estudiar el tiempo que este cada modelo activo