Update app.py

app.py

@@ -34,7 +34,7 @@ def setup_cfg(config_file):
 
 @spaces.GPU
 @torch.no_grad()
-@torch.autocast(device_type="cuda", dtype=torch.float16)
+@torch.autocast(device_type="cuda", dtype=torch.float32)
 def inference_automatic(input_img, class_names):
     mp.set_start_method("spawn", force=True)
     config_file = './configs/ground-truth-warmup/mask-adapter/mask_adapter_convnext_large_cocopan_eval_ade20k.yaml'
@@ -59,7 +59,7 @@ def inference_automatic(input_img, class_names):
 
 @spaces.GPU
 @torch.no_grad()
-@torch.autocast(device_type="cuda", dtype=torch.float16)
+@torch.autocast(device_type="cuda", dtype=torch.float32)
 def inference_point(input_img, evt: gr.SelectData,):
 
     x, y = evt.index[0], evt.index[1]