Upload app.py

app.py

@@ -5,11 +5,12 @@ from transformers import pipeline
 import time
 import torch
 
-generate_story_pipe = pipeline("text2text-generation", model="openai-community/gpt2")
+generate_story_pipe = pipeline("text-generation", model="openai-community/gpt2")
 
 # 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):
     def wrapper(*args, **kwargs):
+        print(f"Starting execution of '{func.__name__}' with 'args={args}, kwargs={kwargs}'")
         start = time.time()
         result = func(*args, **kwargs)
         end = time.time()