Spaces:
Sleeping
Sleeping
test
Browse files
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="
|
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,
|