Update app.py
Browse files
app.py
CHANGED
@@ -11,7 +11,7 @@ import numpy as np
|
|
11 |
|
12 |
|
13 |
args = {
|
14 |
-
"model_path": "model_last_epoch_34_torchvision0_3_state.ptw"
|
15 |
"device": torch.device('cuda' if torch.cuda.is_available() else 'cpu')
|
16 |
}
|
17 |
|
|
|
11 |
|
12 |
|
13 |
args = {
|
14 |
+
"model_path": "model_last_epoch_34_torchvision0_3_state.ptw",
|
15 |
"device": torch.device('cuda' if torch.cuda.is_available() else 'cpu')
|
16 |
}
|
17 |
|