Javi commited on
Commit
d3acf41
·
1 Parent(s): 0256e37

Fixed bug with mocking

Browse files
Files changed (1) hide show
  1. streamlit_app.py +2 -0
streamlit_app.py CHANGED
@@ -290,6 +290,8 @@ col1.markdown("#### Task selection")
290
  task_name: str = col2.selectbox("", options=["Prompt ranking", "Image ranking", "Image classification"])
291
  st.markdown("<br>", unsafe_allow_html=True)
292
 
 
 
293
  session_state = get_state()
294
  if task_name == "Image classification":
295
  Sections.image_uploader(session_state, accept_multiple_files=False)
 
290
  task_name: str = col2.selectbox("", options=["Prompt ranking", "Image ranking", "Image classification"])
291
  st.markdown("<br>", unsafe_allow_html=True)
292
 
293
+ images_mocker.stop_mocking() # Sometimes it gets stuck mocking
294
+
295
  session_state = get_state()
296
  if task_name == "Image classification":
297
  Sections.image_uploader(session_state, accept_multiple_files=False)