sindhuhegde commited on
Commit
e667a31
1 Parent(s): 71b5ee4

Update app

Browse files
Files changed (2) hide show
  1. .DS_Store +0 -0
  2. app.py +1 -1
.DS_Store ADDED
Binary file (6.15 kB). View file
 
app.py CHANGED
@@ -28,7 +28,7 @@ warnings.filterwarnings("ignore", category=DeprecationWarning)
28
  warnings.filterwarnings("ignore", category=UserWarning)
29
 
30
  # Set the path to checkpoint file
31
- CHECKPOINT_PATH = "checkpoints/model_rgb.pth" # Update this path
32
 
33
  # Initialize global variables
34
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
28
  warnings.filterwarnings("ignore", category=UserWarning)
29
 
30
  # Set the path to checkpoint file
31
+ CHECKPOINT_PATH = "model_rgb.pth"
32
 
33
  # Initialize global variables
34
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")