fix model loading bug

app.py

@@ -41,7 +41,7 @@ fusion_policy = 'cat'
 
 # build model
 set_random_seed(random_seed)
-metauas_model = MetaUAS(encoder_name, 
+model = MetaUAS(encoder_name, 
                 decoder_name, 
                 encoder_depth, 
                 decoder_depth, 
@@ -56,7 +56,7 @@ def process_image(prompt_img, query_img, options):
         #model = safely_load_state_dict(metauas_model, ckt_path)
         
         ckpt_path = hf_hub_download(repo_id="csgaobb/MetaUAS", filename="metauas-512.ckpt")
-        model = metauas_model.load_state_dict(torch.load(ckpt_path), strict=True)
+        model.load_state_dict(torch.load(ckpt_path), strict=True)
   
         #model = MetaUAS.from_pretrained("csgaobb/MetaUAS-512")
         img_size = 512
@@ -64,7 +64,7 @@ def process_image(prompt_img, query_img, options):
         #model = safely_load_state_dict(metauas_model, ckt_path)
 
         ckpt_path = hf_hub_download(repo_id="csgaobb/MetaUAS", filename='metauas-256.ckpt')
-        model = metauas_model.load_state_dict(torch.load(ckpt_path), strict=True)
+        model.load_state_dict(torch.load(ckpt_path), strict=True)
         #model = MetaUAS.from_pretrained("csgaobb/MetaUAS-256")
         img_size = 256