Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -1,6 +1,10 @@
|
|
1 |
import gradio as gr
|
2 |
from transformers import pipeline
|
3 |
|
|
|
|
|
|
|
|
|
4 |
pipe = pipeline(model="suno/bark-small")
|
5 |
output = pipe("Hey it's HuggingFace on the phone!")
|
6 |
|
|
|
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 |
pipe = pipeline(model="suno/bark-small")
|
9 |
output = pipe("Hey it's HuggingFace on the phone!")
|
10 |
|