Update app.py

app.py

@@ -239,5 +239,8 @@ if generate_caption_button:
         generate_caption(1, random_image)
             
     except RuntimeError as e:
-        st.write("TRY AGAIN")
+        random_row = validation.sample()
+        random_image = random_row.iloc[0]['image']
+        
+        generate_caption(1, random_image)