Update app.py

app.py

@@ -19,14 +19,11 @@ def predict():
     st.markdown('#### Predicted Captions:')
     captions.append(pred_caption)
 
-    for _ in range(5):
+    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)
-
-        if len(captions)==3:
-            return
     
     for c in captions:
         st.write(c)