BhumikaMak commited on
Commit
c2d0764
·
1 Parent(s): 4365154

Fix: argument mismatch

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -23,7 +23,7 @@ def process_image(image, yolo_versions=["yolov5"]):
23
  interface = gr.Interface(
24
  fn=process_image,
25
  inputs=[
26
- gr.Image(type="pil", label="Upload an Image", optional=True),
27
  gr.CheckboxGroup(
28
  choices=["yolov5", "yolov8s"],
29
  value=["yolov5"], # Set default selection to YOLOv5
@@ -41,5 +41,6 @@ interface = gr.Interface(
41
  )
42
 
43
 
 
44
  if __name__ == "__main__":
45
  interface.launch()
 
23
  interface = gr.Interface(
24
  fn=process_image,
25
  inputs=[
26
+ gr.Image(type="pil", label="Upload an Image"),
27
  gr.CheckboxGroup(
28
  choices=["yolov5", "yolov8s"],
29
  value=["yolov5"], # Set default selection to YOLOv5
 
41
  )
42
 
43
 
44
+
45
  if __name__ == "__main__":
46
  interface.launch()