AlshimaaGamalAlsaied commited on
Commit
86f489c
1 Parent(s): c0892d3
Files changed (3) hide show
  1. app.py +3 -3
  2. best _baseline.pt +3 -0
  3. requirements.txt +3 -0
app.py CHANGED
@@ -1,13 +1,13 @@
1
  import gradio as gr
2
- import torch
3
  import yolov7
4
 
5
 
6
  from huggingface_hub import hf_hub_download
7
 
8
  # Images
9
- torch.hub.download_url_to_file('https://github.com/ultralytics/yolov5/raw/master/data/images/zidane.jpg', 'zidane.jpg')
10
- torch.hub.download_url_to_file('https://raw.githubusercontent.com/obss/sahi/main/tests/data/small-vehicles1.jpeg', 'small-vehicles1.jpeg')
11
 
12
  def yolov7_inference(
13
  image: gr.inputs.Image = None,
 
1
  import gradio as gr
2
+ #import torch
3
  import yolov7
4
 
5
 
6
  from huggingface_hub import hf_hub_download
7
 
8
  # Images
9
+ #torch.hub.download_url_to_file('https://github.com/ultralytics/yolov5/raw/master/data/images/zidane.jpg', 'zidane.jpg')
10
+ #torch.hub.download_url_to_file('https://raw.githubusercontent.com/obss/sahi/main/tests/data/small-vehicles1.jpeg', 'small-vehicles1.jpeg')
11
 
12
  def yolov7_inference(
13
  image: gr.inputs.Image = None,
best _baseline.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9b6af4a8e7deaa99298717dc6a6ddf9bc7a3bf20e83c7d6710543d8d61c945bf
3
+ size 298841412
requirements.txt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ gradio
2
+ torch
3
+ yolov7detect