Update app.py

app.py

@@ -6,13 +6,16 @@ import os
 
 ## Where to store flagged data
 HF_TOKEN = os.getenv('HF_TOKEN')
-hf_writer = gr.HuggingFaceDatasetSaver(HF_TOKEN, "Crowd sourced images")
+hf_writer = gr.HuggingFaceDatasetSaver(HF_TOKEN, "flaggedImages", private=True)
 
 # Load your trained model
 model = YOLO('best.pt')
 
 #Function for making predictions
 def predict (image):
+    ##To display an error message if you submit without an image
+    if image is None:
+        return gr.err("Please upload a chest X-ray image to proceed")
     results = model(image, conf =0.2)
     ## list of all the probabilities
     conf = []