muratt0 commited on
Commit
84f7ae8
·
verified ·
1 Parent(s): 415e8cd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -7,7 +7,7 @@ from datetime import datetime
7
  import os
8
 
9
  # Load the model
10
- model = YOLO('temp_best.pt')
11
 
12
  def preprocess_image(image, blur_amount, brightness_level, rotation_angle):
13
  # Convert to PIL Image if needed
 
7
  import os
8
 
9
  # Load the model
10
+ model = YOLO('best.pt')
11
 
12
  def preprocess_image(image, blur_amount, brightness_level, rotation_angle):
13
  # Convert to PIL Image if needed