Faisal
commited on
Commit
·
488d84d
1
Parent(s):
c1d8c8a
Restore GPU version - remove CPU optimizations and restore GPU-compatible dependencies
Browse files
app.py
CHANGED
@@ -39,7 +39,7 @@ temp_generation_config = GenerationConfig(
|
|
39 |
# ----------------------------
|
40 |
# YOLO MODEL LOADING
|
41 |
# ----------------------------
|
42 |
-
yolo_model = YOLO("
|
43 |
|
44 |
def inference(image_path: str):
|
45 |
"""Runs YOLO segmentation on an image and returns the annotated image."""
|
|
|
39 |
# ----------------------------
|
40 |
# YOLO MODEL LOADING
|
41 |
# ----------------------------
|
42 |
+
yolo_model = YOLO("best.pt") # replace with your segmentation weights
|
43 |
|
44 |
def inference(image_path: str):
|
45 |
"""Runs YOLO segmentation on an image and returns the annotated image."""
|