Muhammed Ömer ERKOÇ commited on
Commit
cf8d6e3
·
1 Parent(s): 255a4ae

Add app.py, requirements.txt, examples and model files 4

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -11,7 +11,7 @@ from lib.pvt import PolypPVT # senin repo'daki model
11
  # ----------------------
12
  # Model yükleme
13
  # ----------------------
14
- pth_path = "weights\PolypPVT.pth"
15
 
16
  model = PolypPVT()
17
  model.load_state_dict(torch.load(pth_path, map_location="cuda" if torch.cuda.is_available() else "cpu"))
 
11
  # ----------------------
12
  # Model yükleme
13
  # ----------------------
14
+ pth_path = "./weights/PolypPVT.pth"
15
 
16
  model = PolypPVT()
17
  model.load_state_dict(torch.load(pth_path, map_location="cuda" if torch.cuda.is_available() else "cpu"))