Spaces:
Sleeping
Sleeping
Muhammed Ömer ERKOÇ
commited on
Commit
·
cf8d6e3
1
Parent(s):
255a4ae
Add app.py, requirements.txt, examples and model files 4
Browse files
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
|
| 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"))
|