Update app.py

app.py

@@ -31,7 +31,7 @@ def process_img(image):
         result = model(y)
             
 
-    restored_img = result.squeeze().permute(1,2,0).clamp_(0, 1).cpu().detach().numpy()
+    restored_img = result.permute(1,2,0).clamp_(0, 1).cpu().detach().numpy()
     restored_img = np.clip(restored_img, 0. , 1.)
 
     restored_img = (restored_img * 255.0).round().astype(np.uint8)  # float32 to uint8