KeerthiVM commited on
Commit
b98f0ab
·
1 Parent(s): f451981
Files changed (1) hide show
  1. app.py +6 -2
app.py CHANGED
@@ -410,8 +410,12 @@ class SkinGPTClassifier:
410
  ])
411
 
412
  def load_models(self):
413
-
414
- self.meta_model = SkinGPT4(vit_checkpoint_path="dermnet_finetuned_version1.pth")
 
 
 
 
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):