Spaces:

Shrey23
/

Image-Captioning

Runtime error

ShreyMehra commited on Nov 21, 2023

Commit

0fc3158

unverified ·

1 Parent(s): 3ac1bdf

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -24,7 +24,7 @@ def load_model():
     return processor, model
-if "model" not in st.session_state:
     processor, model = load_model() #load model
     st.session_state.dict = {}
     st.session_state.dict['processor'] = processor

     return processor, model
+if "dict" not in st.session_state:
     processor, model = load_model() #load model
     st.session_state.dict = {}
     st.session_state.dict['processor'] = processor