Szeyu commited on
Commit
c047835
·
verified ·
1 Parent(s): 37631cc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -10,7 +10,7 @@ from PIL import Image
10
  # Initialize pipelines
11
  @st.cache_resource
12
  def load_pipelines():
13
- captioner = pipeline("image-to-text", model="Salesforce/blip-image-captioning-base")
14
  storyer = pipeline("text-generation", model="aspis/gpt2-genre-story-generation")
15
  tts = pipeline("text-to-speech", model="facebook/mms-tts-eng")
16
  return captioner, storyer, tts
 
10
  # Initialize pipelines
11
  @st.cache_resource
12
  def load_pipelines():
13
+ captioner = pipeline("image-to-text", model="Salesforce/blip-image-captioning-large")
14
  storyer = pipeline("text-generation", model="aspis/gpt2-genre-story-generation")
15
  tts = pipeline("text-to-speech", model="facebook/mms-tts-eng")
16
  return captioner, storyer, tts