Update gradio_app.py

gradio_app.py

@@ -17,7 +17,7 @@ torch.backends.cudnn.benchmark = True
 class MatchStereoDemo:
     def __init__(self):
         self.has_cuda = torch.cuda.is_available()
-        self.device = torch.device('cuda:0') if self.has_cuda else 'cpu'
+        self.device = "cuda" if self.has_cuda else 'cpu'
         self.model = None
         self.current_variant = None
         self.current_mode = None