fix-2

app.py

@@ -6,7 +6,7 @@ from SwinIR.infer import SwinIRDemo
 from sam_diffsr.infer import sam_diffsr_demo
 
 
-def image_infer(img_PIL, progress= gr.Progress(track_tqdm=True)):
+def image_infer(img_PIL, progress=gr.Progress(track_tqdm=True)):
     sam_diffsr_img = sam_diffsr_infer.infer(img_PIL)
     swin_ir_img = swin_ir_infer.infer(img_PIL)
     return sam_diffsr_img, swin_ir_img

sam_diffsr/models_sr/diffusion_sam.py

@@ -52,7 +52,7 @@ class GaussianDiffusion_sam(GaussianDiffusion):
         return model_mean + nonzero_mask * (0.5 * model_log_variance).exp() * noise, x0_pred
 
     @torch.no_grad()
-    def sample(self, img_lr, img_lr_up, shape, sam_mask=None, save_intermediate=False, progress=None):
+    def sample(self, img_lr, img_lr_up, shape, sam_mask=None, save_intermediate=False):
         device = self.betas.device
         b = shape[0]