Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -196,7 +196,7 @@ def main():
|
|
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.
|
200 |
|
201 |
# Initialize the board for FEN (empty rows represented by "8")
|
202 |
board = [["8"] * 8 for _ in range(8)]
|
|
|
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.5)
|
200 |
|
201 |
# Initialize the board for FEN (empty rows represented by "8")
|
202 |
board = [["8"] * 8 for _ in range(8)]
|