hugo flores garcia
commited on
Commit
•
1abec8d
1
Parent(s):
dafca84
fix models
Browse files
app.py
CHANGED
|
@@ -29,9 +29,9 @@ device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
|
| 29 |
|
| 30 |
interface = Interface(
|
| 31 |
device=device,
|
| 32 |
-
coarse_ckpt="models/
|
| 33 |
-
coarse2fine_ckpt="models/
|
| 34 |
-
codec_ckpt="models/
|
| 35 |
)
|
| 36 |
|
| 37 |
# populate the model choices with any interface.yml files in the generated confs
|
|
|
|
| 29 |
|
| 30 |
interface = Interface(
|
| 31 |
device=device,
|
| 32 |
+
coarse_ckpt="models/nesquik/coarse.pth",
|
| 33 |
+
coarse2fine_ckpt="models/nesquik/c2f.pth",
|
| 34 |
+
codec_ckpt="models/nesquik/codec.pth",
|
| 35 |
)
|
| 36 |
|
| 37 |
# populate the model choices with any interface.yml files in the generated confs
|