Update app.py

app.py

@@ -6,9 +6,6 @@ from tqdm import tqdm
 
 yolo = YOLO()
 
-img_folder = "img/"
-img_files = [os.path.join(img_folder, file) for file in os.listdir(img_folder) if file.endswith(('.png', '.jpg', '.jpeg'))]
-
 def predict_single_image(image, crop=False, count=True):
     try:
         r_image = yolo.detect_image(image, crop=crop, count=count)
@@ -16,25 +13,10 @@ def predict_single_image(image, crop=False, count=True):
     except Exception as e:
         return str(e)
 
-def select_image(image_path):
-    image = Image.open(image_path)
-    return image
-
-# with gr.Blocks() as iface:
-#     with gr.Row():
-#         with gr.Column():
-#             selected_image = gr.Image(label="Selected Image", width=600, height=400)
-#             img_selector = gr.Dropdown(choices=img_files, label="Select an image from folder")
-#             img_selector.change(select_image, img_selector, selected_image)
-#         with gr.Column():
-#             output_image = gr.Image(label="Output Image", width=600, height=400)
-#             predict_button = gr.Button("Predict")
-#             predict_button.click(predict_single_image, inputs=selected_image, outputs=output_image)
-            
 demo = gr.Interface(
     fn=predict_single_image,
     inputs=[
-        gr.Image(value="img/1.png", type="pil", label="Input"),
+        gr.Image(value="img/1.png", type="pil", label="Input Images"),
     ],
 	
     outputs=gr.Image(label="Detection Images",