Spaces:
Sleeping
Sleeping
Update identification_model.py
Browse files- identification_model.py +2 -2
identification_model.py
CHANGED
|
@@ -10,8 +10,8 @@ def load_yolov8_model():
|
|
| 10 |
|
| 11 |
def run_object_detection(model, image_path):
|
| 12 |
# Load the image
|
| 13 |
-
image = Image.open(image_path).convert('RGB')
|
| 14 |
-
image_np = np.array(
|
| 15 |
|
| 16 |
# Run inference
|
| 17 |
results = model(image_np)
|
|
|
|
| 10 |
|
| 11 |
def run_object_detection(model, image_path):
|
| 12 |
# Load the image
|
| 13 |
+
# image = Image.open(image_path).convert('RGB')
|
| 14 |
+
image_np = np.array(image_path)
|
| 15 |
|
| 16 |
# Run inference
|
| 17 |
results = model(image_np)
|