MaxGab commited on
Commit
412c966
·
verified ·
1 Parent(s): b067095

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -5
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
- pipe = pipeline(model="suno/bark-small")
9
- output = pipe("Hey it's HuggingFace on the phone!")
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"]