delete duplicated images and adding removing gpu dependency

app.py

@@ -11,6 +11,9 @@ class VSGradio:
     def __init__(self, model_config, model_ckpt_path):
         self.model_config = model_config
         self.model_ckpt_path = model_ckpt_path
+        self.device = torch.device(
+            "cuda" if torch.cuda.is_available() else "cpu"
+        )  # Check if GPU is available
         self.model = None
         self.load_model()
 
@@ -20,10 +23,9 @@ class VSGradio:
             self.model_ckpt_path,
             architecture="UNeXt2_2D",
             model_config=self.model_config,
-            accelerator="gpu",
         )
+        self.model.to(self.device)
         self.model.eval()
-        self.model
 
     def normalize_fov(self, input: ArrayLike):
         "Normalizing the fov with zero mean and unit variance"
@@ -38,7 +40,7 @@ class VSGradio:
         inp = torch.from_numpy(np.array(inp).astype(np.float32))
         test_dict = dict(
             index=None,
-            source=inp.unsqueeze(0).unsqueeze(0).unsqueeze(0).to(self.model.device),
+            source=inp.unsqueeze(0).unsqueeze(0).unsqueeze(0).to(self.device),
         )
         with torch.inference_mode():
             self.model.on_predict_start()