Update app.py

app.py

@@ -42,6 +42,7 @@ class model:
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
  if self.model_name != model_name:
  self.model_name = model_name
+ del self.model
  if self.model_name not in self.model_dict.keys():
  model_ckpt_path = hf_hub_download(repo_id=f"HuangLab/{model_name}", filename="model.ckpt")
  model_config_path = hf_hub_download(repo_id=f"HuangLab/{model_name}", filename="config.yaml")