Fix
Browse files
app.py
CHANGED
@@ -410,8 +410,12 @@ class SkinGPTClassifier:
|
|
410 |
])
|
411 |
|
412 |
def load_models(self):
|
413 |
-
|
414 |
-
|
|
|
|
|
|
|
|
|
415 |
self.meta_model.to_empty(device=device)
|
416 |
|
417 |
def predict(self, image, top_k=3):
|
|
|
410 |
])
|
411 |
|
412 |
def load_models(self):
|
413 |
+
model_path = hf_hub_download(
|
414 |
+
repo_id="KeerthiVM/SkinCancerDiagnosis",
|
415 |
+
filename="dermnet_finetuned_version1.pth",
|
416 |
+
)
|
417 |
+
# self.meta_model = SkinGPT4(vit_checkpoint_path="dermnet_finetuned_version1.pth")
|
418 |
+
self.meta_model = SkinGPT4(vit_checkpoint_path=model_path)
|
419 |
self.meta_model.to_empty(device=device)
|
420 |
|
421 |
def predict(self, image, top_k=3):
|