hskwon7 commited on
Commit
2a4516f
·
verified ·
1 Parent(s): cdc9632

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -9,7 +9,7 @@ st.write("Upload an image and watch as it’s captioned, turned into a short sto
9
 
10
  @st.cache_resource
11
  def load_captioner():
12
- return pipeline("image-captioning", model="nlpconnect/vit-gpt2-image-captioning")
13
 
14
  @st.cache_resource
15
  def load_story_gen():
 
9
 
10
  @st.cache_resource
11
  def load_captioner():
12
+ return pipeline("image-to-text", model="unography/blip-large-long-cap")
13
 
14
  @st.cache_resource
15
  def load_story_gen():