RafaelJaime commited on
Commit
7b981ed
·
verified ·
1 Parent(s): a3311b3

Upload app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -5,7 +5,7 @@ from transformers import pipeline
5
  import time
6
  import torch
7
 
8
- generate_story_pipe = pipeline("text-generation", model="Qwen/Qwen2.5-1.5B-Instruct")
9
 
10
  # 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
11
  def measure_performance(func):
 
5
  import time
6
  import torch
7
 
8
+ generate_story_pipe = pipeline("text-generation", model="Qwen/Qwen2.5-0.5B-Instruct")
9
 
10
  # 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
11
  def measure_performance(func):