Spaces:

alshimaa
/

Mask_detection

Runtime error

AlshimaaGamalAlsaied commited on Apr 19, 2023

Commit

a39abfb

1 Parent(s): 8e4ec6f

commit

Files changed (1) hide show

app.py CHANGED Viewed

@@ -33,7 +33,7 @@ def yolov5_inference(
 inputs = [
     gr.inputs.Image(type="pil", label="Input Image"),
-    gr.inputs.Dropdown(["yolov5s.pt", "alshimaa/yolo5_epoch100"], label="Model"),
     gr.inputs.Slider(minimum=320, maximum=1280, default=640, step=32, label="Image Size"),
     gr.inputs.Slider(minimum=0.0, maximum=1.0, default=0.25, step=0.05, label="Confidence Threshold"),
     gr.inputs.Slider(minimum=0.0, maximum=1.0, default=0.45, step=0.05, label="IOU Threshold"),

 inputs = [
     gr.inputs.Image(type="pil", label="Input Image"),
+    gr.inputs.Dropdown(["yolov5s.pt", "yolov5l.pt", "yolov5x.pt"], label="Model"),
     gr.inputs.Slider(minimum=320, maximum=1280, default=640, step=32, label="Image Size"),
     gr.inputs.Slider(minimum=0.0, maximum=1.0, default=0.25, step=0.05, label="Confidence Threshold"),
     gr.inputs.Slider(minimum=0.0, maximum=1.0, default=0.45, step=0.05, label="IOU Threshold"),