Emaad commited on
Commit
8e4e91e
1 Parent(s): 658d610

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -43,6 +43,7 @@ class model:
43
  if self.model_name != model_name:
44
  if self.model_path is not None:
45
  os.remove(self.model_path)
 
46
  self.model_name = model_name
47
  model_ckpt_path = hf_hub_download(repo_id=f"HuangLab/{model_name}", filename="model.ckpt")
48
  self.model_path = model_ckpt_path
 
43
  if self.model_name != model_name:
44
  if self.model_path is not None:
45
  os.remove(self.model_path)
46
+ del self.model
47
  self.model_name = model_name
48
  model_ckpt_path = hf_hub_download(repo_id=f"HuangLab/{model_name}", filename="model.ckpt")
49
  self.model_path = model_ckpt_path