okrakun commited on
Commit
82e5234
·
verified ·
1 Parent(s): a11a512

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -5,9 +5,9 @@ import numpy as np
5
  import tempfile
6
 
7
  # Load models
8
- model_swelling = YOLO("models/swelling/best.pt")
9
- model_redness = YOLO("models/redness/best.pt")
10
- model_bleeding = YOLO("models/bleeding/best.pt")
11
 
12
  def save_temp_file(img):
13
  """Save PIL image to a temporary file and return the file path."""
 
5
  import tempfile
6
 
7
  # Load models
8
+ model_swelling = YOLO("models/swelling/swelling_final.pt")
9
+ model_redness = YOLO("models/redness/redness_final.pt")
10
+ model_bleeding = YOLO("models/bleeding/bleeding_final.pt")
11
 
12
  def save_temp_file(img):
13
  """Save PIL image to a temporary file and return the file path."""