RafaelJaime commited on
Commit
5fe0e11
1 Parent(s): 7b981ed

Upload app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -1,11 +1,12 @@
1
  import gradio as gr
2
  import numpy as np
3
  from bark import SAMPLE_RATE, generate_audio
4
- from transformers import pipeline
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):
 
1
  import gradio as gr
2
  import numpy as np
3
  from bark import SAMPLE_RATE, generate_audio
4
+ from transformers import pipeline, set_seed
5
  import time
6
  import torch
7
 
8
+ generate_story_pipe = pipeline("text-generation", model="openai-community/gpt2")
9
+ set_seed(42)
10
 
11
  # 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
12
  def measure_performance(func):