Spaces:

Harthika2004
/

Image-Caption-Generator-streamlit

Sleeping

MANIKANDAN A commited on Aug 30, 2023

Commit

f540753

1 Parent(s): c9c0096

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -97,7 +97,7 @@ def login_section():
 def predict(cap_col):
     captions = []
-    pred_caption = generate_caption('tmp.jpg', caption_model)
     cap_col.markdown('#### Predicted Captions:')
     captions.append(pred_caption)

 def predict(cap_col):
     captions = []
+    pred_caption = generate_caption('tmp.jpg', get_caption_model)
     cap_col.markdown('#### Predicted Captions:')
     captions.append(pred_caption)