jayn95 commited on
Commit
82ad09c
·
verified ·
1 Parent(s): 3323e08

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -5,9 +5,9 @@ import numpy as np
5
  import tempfile, io, base64
6
 
7
  # Load models
8
- model_swelling = YOLO("models/swelling/best.pt")
9
  model_redness = YOLO("models/redness/redness_final.pt")
10
- model_bleeding = YOLO("models/bleeding/best.pt")
11
 
12
 
13
  # --- Helpers ---
 
5
  import tempfile, io, base64
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
 
13
  # --- Helpers ---