AlshimaaGamalAlsaied commited on
Commit
57d8a9d
Β·
1 Parent(s): a47293c
Files changed (2) hide show
  1. app.py +2 -2
  2. requirements.txt +1 -1
app.py CHANGED
@@ -1,5 +1,5 @@
1
  import gradio as gr
2
- #import torch
3
  import yolov5
4
  import subprocess
5
  import tempfile
@@ -59,7 +59,7 @@ image_interface = gr.Interface(
59
  #gr.inputs.Slider(minimum=0.0, maximum=1.0, default=0.45, step=0.05, label="IOU Threshold")
60
  ],
61
  outputs=gr.outputs.Image(type="filepath", label="Output Image"),
62
- title="Obiect detection model p",
63
  examples=[['img1.png', 'alshimaa/yolo5_epoch100', 640, 0.25, 0.45], ['img2.png', 'alshimaa/yolo5_epoch100', 640, 0.25, 0.45], ['img3.png', 'alshimaa/yolo5_epoch100', 640, 0.25, 0.45]],
64
  cache_examples=True,
65
  theme='huggingface',
 
1
  import gradio as gr
2
+ import torch
3
  import yolov5
4
  import subprocess
5
  import tempfile
 
59
  #gr.inputs.Slider(minimum=0.0, maximum=1.0, default=0.45, step=0.05, label="IOU Threshold")
60
  ],
61
  outputs=gr.outputs.Image(type="filepath", label="Output Image"),
62
+ title="Object Detector: Identify People Without Mask",
63
  examples=[['img1.png', 'alshimaa/yolo5_epoch100', 640, 0.25, 0.45], ['img2.png', 'alshimaa/yolo5_epoch100', 640, 0.25, 0.45], ['img3.png', 'alshimaa/yolo5_epoch100', 640, 0.25, 0.45]],
64
  cache_examples=True,
65
  theme='huggingface',
requirements.txt CHANGED
@@ -1,4 +1,4 @@
1
  gradio
2
  torch
3
- yolov5detect
4
  HfApi
 
1
  gradio
2
  torch
3
+ yolov5
4
  HfApi