Update app.py

app.py

@@ -4,14 +4,12 @@ import os
 from yolo import YOLO
 from tqdm import tqdm
 
-# 初始化YOLO模型
+
 yolo = YOLO()
 
-# 获取 img 文件夹中的图片列表
-img_folder = "img"
+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)
@@ -19,23 +17,20 @@ 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
 
-# 使用Gradio创建界面
 with gr.Blocks() as iface:
     with gr.Row():
         with gr.Column():
-            selected_image = gr.Image(label="Selected Image", width=400, height=400)
+            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=800, height=800)
+            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)
 
-# 启动应用
 if __name__ == "__main__":
     iface.launch()