Hugging Face's logo

Spaces:
momererkoc
/
polyp_segmentation
Sleeping

App Files Community
Muhammed Ömer ERKOÇ commited on
Commit
255a4ae
·
1 Parent(s): 232e523

3

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 = "polyp_segmentation\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"))