siddharth060104 commited on
Commit
7386c79
·
verified ·
1 Parent(s): 65430ae

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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.6)
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)]