chaouch commited on
Commit
745db4e
·
1 Parent(s): 494ccea
Files changed (1) hide show
  1. app.py +8 -1
app.py CHANGED
@@ -7,6 +7,7 @@ import requests
7
 
8
  from yolo import Yolo
9
  model_link = "https://intranet-projects-files.s3.amazonaws.com/holbertonschool-ml/yolo.h5"
 
10
 
11
  def download_model():
12
  if not os.path.exists('data'):
@@ -20,6 +21,12 @@ def download_model():
20
  print("Model downnloaded")
21
  else:
22
  print("Model already exists locally.")
 
 
 
 
 
 
23
  def run(img):
24
  np.random.seed(0)
25
 
@@ -53,7 +60,7 @@ def run(img):
53
 
54
  return img
55
 
56
- demo = gr.Interface(run, "image", "image")
57
 
58
  if __name__ == "__main__":
59
  download_model()
 
7
 
8
  from yolo import Yolo
9
  model_link = "https://intranet-projects-files.s3.amazonaws.com/holbertonschool-ml/yolo.h5"
10
+ im_link = "https://miro.medium.com/v2/resize:fit:720/format:webp/1*EYFejGUjvjPcc4PZTwoufw.jpeg"
11
 
12
  def download_model():
13
  if not os.path.exists('data'):
 
21
  print("Model downnloaded")
22
  else:
23
  print("Model already exists locally.")
24
+ if not os.path.exists('data/exm.jpg'):
25
+ print("Downloading image...")
26
+ r = requests.get(im_link)
27
+ with open('data/exm.jpg', 'wb') as f:
28
+ f.write(r.content)
29
+ print("Image downnloaded")
30
  def run(img):
31
  np.random.seed(0)
32
 
 
60
 
61
  return img
62
 
63
+ demo = gr.Interface(run, "image", "image" , examples=[["data/exm.jpg"]], capture_session=True)
64
 
65
  if __name__ == "__main__":
66
  download_model()