fix model loading bug

app.py

@@ -53,20 +53,18 @@ metauas_model = MetaUAS(encoder_name,
 def process_image(prompt_img, query_img, options):
     # Load the model based on selected options
     if 'model-512' in options:
-        ckt_path = "weights/metauas-512.ckpt"
         #model = safely_load_state_dict(metauas_model, ckt_path)
         
-        ckpt_path = hf_hub_download(repo_id="csgaobb/MetaUAS", filename=ckt_path)
-        metauas_model.load_state_dict(torch.load(ckpt_path))
+        ckpt_path = hf_hub_download(repo_id="csgaobb/MetaUAS", filename="metauas-512.ckpt")
+        model = metauas_model.load_state_dict(torch.load(ckpt_path))
   
         #model = MetaUAS.from_pretrained("csgaobb/MetaUAS-512")
         img_size = 512
     else:
-        ckt_path = 'weights/metauas-256.ckpt'
         #model = safely_load_state_dict(metauas_model, ckt_path)
 
-        ckpt_path = hf_hub_download(repo_id="csgaobb/MetaUAS", filename=ckt_path)
-        metauas_model.load_state_dict(torch.load(ckpt_path))
+        ckpt_path = hf_hub_download(repo_id="csgaobb/MetaUAS", filename='metauas-256.ckpt')
+        model = metauas_model.load_state_dict(torch.load(ckpt_path))
         #model = MetaUAS.from_pretrained("csgaobb/MetaUAS-256")
         img_size = 256