fix gpu error

app.py

@@ -85,7 +85,8 @@ def process_image(args, inpainting_pipeline, net):
         img_with_green_bg = img
     image_tensor = transform_rgb(img_with_green_bg)
     image_tensor = image_tensor.unsqueeze(0)
-    output_tensor = net(image_tensor.to(device))
+    with torch.autocast(device_type=device):
+        output_tensor = net(image_tensor.to(device))
     output_tensor = F.log_softmax(output_tensor[0], dim=1)
     output_tensor = torch.max(output_tensor, dim=1, keepdim=True)[1]
     output_tensor = torch.squeeze(output_tensor, dim=0)