Spaces:
Running
Running
Commit ·
49796a8
1
Parent(s): 7c2e2bd
typo fixed
Browse files
app.py
CHANGED
|
@@ -19,7 +19,7 @@ from mobileclip.modules.common.mobileone import reparameterize_model
|
|
| 19 |
from model import MobileCLIPRanker
|
| 20 |
|
| 21 |
HF_USER_REPO = "Nightfury16/clipick"
|
| 22 |
-
HF_FILENAME = "
|
| 23 |
CONFIG_PATH = "config.yml"
|
| 24 |
JSON_DATA_PATH = "combined_unique.json"
|
| 25 |
DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
|
|
|
|
| 19 |
from model import MobileCLIPRanker
|
| 20 |
|
| 21 |
HF_USER_REPO = "Nightfury16/clipick"
|
| 22 |
+
HF_FILENAME = "best_student_b.pth"
|
| 23 |
CONFIG_PATH = "config.yml"
|
| 24 |
JSON_DATA_PATH = "combined_unique.json"
|
| 25 |
DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
|