MaxGab commited on
Commit
c7bc873
·
verified ·
1 Parent(s): 6e97f3b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -0
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