Update app.py

app.py

@@ -587,7 +587,8 @@ def infer_upscale(
         if control_image.mode != 'RGB':
             control_image = control_image.convert('RGB')
         
-        control_image = torch.from_numpy(np.array(control_image)).permute(2, 0, 1).float().to(device) / 255.0
+        # Convert to tensor and add batch dimension
+        control_image = torch.from_numpy(np.array(control_image)).permute(2, 0, 1).float().unsqueeze(0).to(device) / 255.0
         
         image = pipe_upscale(
             prompt="",