Update app.py

app.py

@@ -35,8 +35,8 @@ def run(image, mode, sample_c=1.3,  num_samples=3, sample_step=100):
     if mode == 'sketch':
         args.mode = 'coco-edge'
         args_up.mode = 'coco-edge'
-        args.model_path =  hf_hub_download(repo_id="tfwang/PITI", filename="base_edge.pt")
-        args.sr_model_path = hf_hub_download(repo_id="tfwang/PITI", filename="upsample_edge.pt")
+        args.model_path =  hf_hub_download(repo_id="tfwang/PITI", filename="base.pt")
+        args.sr_model_path = hf_hub_download(repo_id="tfwang/PITI", filename="upsample.pt")
 
     elif mode == 'mask':
         args.mode = 'coco'