Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -193,7 +193,7 @@ def main():
|
|
193 |
height, width, _ = processed_image.shape
|
194 |
|
195 |
# Initialize the YOLO model
|
196 |
-
model = YOLO("
|
197 |
|
198 |
# Run detection
|
199 |
results = model.predict(source=processed_image, save=False, save_txt=False, conf=0.6)
|
|
|
193 |
height, width, _ = processed_image.shape
|
194 |
|
195 |
# Initialize the YOLO model
|
196 |
+
model = YOLO("fine_tuned_on_all_data.pt") # Replace with your trained model weights file
|
197 |
|
198 |
# Run detection
|
199 |
results = model.predict(source=processed_image, save=False, save_txt=False, conf=0.6)
|