clean up app
Browse files
app.py
CHANGED
|
@@ -8,10 +8,9 @@ from torchvision import transforms
|
|
| 8 |
REPO_ID = "Launchpad/ditto"
|
| 9 |
FILENAME = "model.pth"
|
| 10 |
|
| 11 |
-
# model_dict = torch.load("model.pth")
|
| 12 |
model_dict = torch.load(
|
| 13 |
-
|
| 14 |
-
|
| 15 |
generator = define_G(input_nc=3, output_nc=3, ngf=64, netG="resnet_9blocks", norm="instance")
|
| 16 |
generator.load_state_dict(model_dict)
|
| 17 |
generator.eval()
|
|
|
|
| 8 |
REPO_ID = "Launchpad/ditto"
|
| 9 |
FILENAME = "model.pth"
|
| 10 |
|
|
|
|
| 11 |
model_dict = torch.load(
|
| 12 |
+
hf_hub_download(repo_id=REPO_ID, filename=FILENAME)
|
| 13 |
+
)
|
| 14 |
generator = define_G(input_nc=3, output_nc=3, ngf=64, netG="resnet_9blocks", norm="instance")
|
| 15 |
generator.load_state_dict(model_dict)
|
| 16 |
generator.eval()
|