JSY8 commited on
Commit
81d0728
·
verified ·
1 Parent(s): d7cbf96

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -14,7 +14,7 @@ import re
14
  def load_pipelines():
15
  captioner = pipeline("image-to-text", model="Salesforce/blip-image-captioning-base")
16
  storyer = pipeline("text-generation", model="gpt2")
17
- tts = pipeline("text-to-audio", model=facebook/musicgen-small)
18
  return captioner, storyer, tts
19
 
20
  captioner, storyer, tts = load_pipelines()
 
14
  def load_pipelines():
15
  captioner = pipeline("image-to-text", model="Salesforce/blip-image-captioning-base")
16
  storyer = pipeline("text-generation", model="gpt2")
17
+ tts = pipeline("text-to-audio", model="facebook/musicgen-small")
18
  return captioner, storyer, tts
19
 
20
  captioner, storyer, tts = load_pipelines()