szeandlinkProject_Testing

Sleeping

Szeyu commited on Apr 30

Commit

b540ff3

verified ·

1 Parent(s): 2e8ed85

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -45,7 +45,7 @@ def load_models():
     if "captioner" not in st.session_state:
         st.session_state.captioner = pipeline(
             "image-to-text",
-            model="Salesforce/blip-image-captioning-base"
         )
     if "storyer" not in st.session_state:
         st.session_state.storyer = pipeline(

     if "captioner" not in st.session_state:
         st.session_state.captioner = pipeline(
             "image-to-text",
+            model="Salesforce/blip-image-captioning-large"
         )
     if "storyer" not in st.session_state:
         st.session_state.storyer = pipeline(