Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -1,12 +1,9 @@
|
|
1 |
import gradio as gr
|
2 |
from transformers import pipeline
|
3 |
|
4 |
-
if tokenizer.pad_token is None:
|
5 |
-
tokenizer.pad_token = tokenizer.eos_token
|
6 |
-
model.config.pad_token_id = model.config.eos_token_id
|
7 |
|
8 |
-
|
9 |
-
|
10 |
|
11 |
audio = output["audio"]
|
12 |
sampling_rate = output["sampling_rate"]
|
|
|
1 |
import gradio as gr
|
2 |
from transformers import pipeline
|
3 |
|
|
|
|
|
|
|
4 |
|
5 |
+
|
6 |
+
pipe = pipeline(model="facebook/musicgen-small")
|
7 |
|
8 |
audio = output["audio"]
|
9 |
sampling_rate = output["sampling_rate"]
|