Spaces:
Running
on
Zero
Running
on
Zero
sindhuhegde
commited on
Commit
•
e667a31
1
Parent(s):
71b5ee4
Update app
Browse files
.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 = "
|
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")
|