Update app.py
Browse files
app.py
CHANGED
|
@@ -28,7 +28,7 @@ css = """
|
|
| 28 |
height: 62px;
|
| 29 |
}
|
| 30 |
"""
|
| 31 |
-
DEVICE = 'cuda' if torch.cuda.is_available() else '
|
| 32 |
config = 'panda_large.yaml'
|
| 33 |
with open(config, 'r') as f:
|
| 34 |
config = yaml.load(f, Loader=yaml.FullLoader)
|
|
|
|
| 28 |
height: 62px;
|
| 29 |
}
|
| 30 |
"""
|
| 31 |
+
DEVICE = 'cuda' if torch.cuda.is_available() else 'cpu'
|
| 32 |
config = 'panda_large.yaml'
|
| 33 |
with open(config, 'r') as f:
|
| 34 |
config = yaml.load(f, Loader=yaml.FullLoader)
|