Update app.py

app.py

@@ -7,26 +7,16 @@ from model import bone_frac
 
 
 @st.cache(allow_output_mutation=True)
-# def get_model():
-#     return get_caption_model()
+def get_model():
+    return get_caption_model()
 
-# caption_model = get_model()
+pred_model = get_model()
 # pred_model=bone_frac()
 
-# def predict():
-#     captions = []
-#     pred_caption = generate_caption('tmp.jpg', caption_model)
-
-#     st.markdown('#### Predicted Captions:')
-#     captions.append(pred_caption)
-
-#     for _ in range(4):
-#         pred_caption = generate_caption('tmp.jpg', caption_model, add_noise=True)
-#         if pred_caption not in captions:
-#             captions.append(pred_caption)
-    
-#     for c in captions:
-        # st.write(c)
+def predict():
+    c=pred_model('tmp.jpg')
+    st.markdown('#### Predicted Captions:')
+    st.write(c)
 
 st.title('Image Captioner')
 img_url = st.text_input(label='Enter Image URL')