Upload app.py

app.py

@@ -5,9 +5,7 @@ from transformers import pipeline
 import time
 import torch
 
-device = "cuda" if torch.cuda.is_available() else "cpu"
 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
 def measure_performance(func):