JSY8 commited on
Commit
759d481
·
verified ·
1 Parent(s): 5b2c907

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -13,7 +13,7 @@ import string
13
  @st.cache_resource
14
  def load_pipelines():
15
  captioner = pipeline("image-to-text", model="Salesforce/blip-image-captioning-large")
16
- storyer = pipeline("text-generation", model="wy2001/storygeneratorllama3.21b")
17
  tts = pipeline("text-to-speech", model="facebook/mms-tts-eng")
18
  return captioner, storyer, tts
19
 
 
13
  @st.cache_resource
14
  def load_pipelines():
15
  captioner = pipeline("image-to-text", model="Salesforce/blip-image-captioning-large")
16
+ storyer = pipeline("text-generation", model="ajibawa-2023/Young-Children-Storyteller-Mistral-7B")
17
  tts = pipeline("text-to-speech", model="facebook/mms-tts-eng")
18
  return captioner, storyer, tts
19