update: add dropddown input to choose model

app.py

@@ -3,16 +3,24 @@ from PIL import Image
 import torch
 from ultralyticsplus import YOLO, render_result
 
+available_models = ["YOLOv8n", "YOLOv8n-GhostNet-P5", "YOLOv8n-GhostNet-P6"]
+available_models_path = [
+    "./models/yolov8n.pt",
+    "./models/yolov8n_ghostnet_p5.pt",
+    "./models/yolov8n_ghostnet_p6.pt",
+]
+
 
 def launch(
     image: gr.Image = None,
-    image_size: gr.Slider = 640,
+    selectedModel: gr.Dropdown = available_models[0],
     conf_threshold: gr.Slider = 0.4,
     iou_threshold: gr.Slider = 0.50,
 ):
+    selected_model_index = available_models.index(selectedModel)
+    image_size = (256,)
     try:
-        model_path = "./models/student-behaviour-best.pt"
-        model = YOLO("./models/OWN-DATASET-640-e120-b32-best.pt")
+        model = YOLO(available_models_path[selected_model_index])
 
         # pil_image = Image.fromarray(image)
 
@@ -31,7 +39,13 @@ def launch(
 
 inputs = [
     gr.Image(type="filepath", label="Input Image"),
-    gr.Slider(minimum=256, maximum=1280, value=640, step=32, label="Image Size"),
+    gr.Dropdown(
+        info="Choose which model should be used in this task",
+        choices=available_models,
+        value=available_models[0],
+        label="Models",
+    ),
+    # gr.Slider(minimum=256, maximum=1280, value=640, step=32, label="Image Size"),
     gr.Slider(
         minimum=0.0, maximum=1.0, value=0.4, step=0.1, label="Confidence Threshold"
     ),

flagged/log.csv

@@ -0,0 +1,2 @@
+Input Image,Image Size,Confidence Threshold,IOU Threshold,Output Result,flag,username,timestamp
+flagged\Input Image\371f0f6a8d35c7500671\clipboard.png,640,0.4,0.4,flagged\Output Result\978dc4411b634e1e2eee\image.png,,,2024-04-24 16:40:16.164367

models/yolov8n.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:eb8d2f237d31ec794d964e2bda2ab20b28f50341616745fed050387b133557f4
+size 6213465

models/yolov8n_ghostnet_p5.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:cbb3b57aef66d4b551b9211893a46006a3aec5983651c872de5843babadd6cfe
+size 5764579

models/yolov8n_ghostnet_p6.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:746b714ff28c4b5cf561c9aeb512f2620697100e7b3257dadbea77439a525f3a
+size 3718685