Update tab_manager.py

tab_manager.py

@@ -328,20 +328,24 @@ def initiate_task_oriented_bias_evaluation(tab, modelID, specs, target, imagesTa
     TASKImages = []
     TASKCaptions = []
     with tab:
+        captionsToExtract = 50
         st.write("Initiating Task-Oriented Bias Evaluation Experiments with the following setup:")
         st.write(" ***Model*** = ", modelID)
         infoColumn1, infoColumn2 = st.columns(2)
+        st.write(" ***No. prompts (pre-defined)*** = ", captionsToExtract)
         st.write(" ***No. Images per prompt*** = ", specs["TO Values"][0])
         st.write(" ***No. Steps*** = ", specs["TO Values"][1])
         st.write(" ***Image Size*** = ", specs["TO Values"][2], "$\\times$", specs["TO Values"][3])
         st.write(" ***Target*** = ", target.lower())
         st.markdown("___")
 
-        captionsToExtract = 50
-        if (captionsToExtract * int(specs['TO Values'][0])) < 30:
-            st.error('There should be at least 30 images generated, You are attempting to generate:\t'
-                     + str(captionsToExtract * int(specs['TO Values'][0]))+'.\nPlease readjust your No. Images per prompt',
+        
+        if int(specs['TO Values'][0]) < 1:
+            st.error('Please readjust your No. Images per prompt to be a valid number (1 or greater)',
                      icon="🚨")
+            # st.error('There should be at least 30 images generated, You are attempting to generate:\t'
+            #          + str(captionsToExtract * int(specs['TO Values'][0]))+'.\nPlease readjust your No. Images per prompt',
+            #          icon="🚨")
         else:
             COCOLoadingBar = st.progress(0, text="Scanning through COCO Dataset for relevant prompts. Please wait")
             prompts, cocoIDs = get_COCO_captions('data/COCO_captions.json', target.lower(), COCOLoadingBar, captionsToExtract)