jonas commited on
Commit
437cc5e
1 Parent(s): 7925eaf
Files changed (1) hide show
  1. app.py +1 -3
app.py CHANGED
@@ -3,11 +3,9 @@ import cv2
3
  import requests
4
  import os
5
  from PIL import Image
6
-
7
  import torch
8
  import ultralytics
9
 
10
-
11
  model = torch.hub.load("ultralytics/yolov5", "custom", path="yolov5_0.65map_exp7_best.pt",
12
  force_reload=False)
13
 
@@ -52,7 +50,7 @@ inputs_image = [
52
  gr.components.Image(type="filepath", label="Input Image"),
53
  ]
54
  outputs_image = [
55
- gr.components.Image(type="file", label="Output Image"),
56
  ]
57
  interface_image = gr.Interface(
58
  fn=show_preds_image,
 
3
  import requests
4
  import os
5
  from PIL import Image
 
6
  import torch
7
  import ultralytics
8
 
 
9
  model = torch.hub.load("ultralytics/yolov5", "custom", path="yolov5_0.65map_exp7_best.pt",
10
  force_reload=False)
11
 
 
50
  gr.components.Image(type="filepath", label="Input Image"),
51
  ]
52
  outputs_image = [
53
+ gr.components.Image(type="filepath", label="Output Image"),
54
  ]
55
  interface_image = gr.Interface(
56
  fn=show_preds_image,