Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -95,7 +95,7 @@ def main():
|
|
95 |
f.write(image_file.getbuffer())
|
96 |
|
97 |
# Load the YOLO models
|
98 |
-
model = YOLO("chessDetection3d") # Replace with your trained model weights file
|
99 |
|
100 |
|
101 |
seg_model = YOLO("segmentation.pt")
|
|
|
95 |
f.write(image_file.getbuffer())
|
96 |
|
97 |
# Load the YOLO models
|
98 |
+
model = YOLO("chessDetection3d.pt") # Replace with your trained model weights file
|
99 |
|
100 |
|
101 |
seg_model = YOLO("segmentation.pt")
|