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()
|