Spaces:
Running
Running
RafaelJaime
commited on
Commit
•
5fe0e11
1
Parent(s):
7b981ed
Upload app.py
Browse files
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="
|
|
|
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):
|