Spaces:
Running
Running
Fix bug and new samples
Browse files
app.py
CHANGED
@@ -10,9 +10,6 @@ def handle_classify(image=None):
|
|
10 |
Args:
|
11 |
image (gr.inputs.Image, optional): Input image to detect objects on. Defaults to None.
|
12 |
"""
|
13 |
-
|
14 |
-
if not image:
|
15 |
-
return "No image found"
|
16 |
|
17 |
model_path = "racist2.0.pt"
|
18 |
model = YOLO(model_path)
|
|
|
10 |
Args:
|
11 |
image (gr.inputs.Image, optional): Input image to detect objects on. Defaults to None.
|
12 |
"""
|
|
|
|
|
|
|
13 |
|
14 |
model_path = "racist2.0.pt"
|
15 |
model = YOLO(model_path)
|