Spaces:
Sleeping
Sleeping
Commit
·
cc28da2
1
Parent(s):
79889bc
new model
Browse files
app.py
CHANGED
@@ -71,7 +71,7 @@ class StrongGenerator(nn.Module):
|
|
71 |
generator = StrongGenerator().to(device)
|
72 |
|
73 |
|
74 |
-
generator.load_state_dict(torch.load("./
|
75 |
generator.eval()
|
76 |
|
77 |
def restore_image(mosaic_image):
|
|
|
71 |
generator = StrongGenerator().to(device)
|
72 |
|
73 |
|
74 |
+
generator.load_state_dict(torch.load("./generator_epoch_100.pth", map_location=device))
|
75 |
generator.eval()
|
76 |
|
77 |
def restore_image(mosaic_image):
|
generator_epoch_70.pth → generator_epoch_100.pth
RENAMED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
-
size
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:926b6b12ad992cb2b4b1fc835f08d7f6c7ec881d76598eca80d592081790b553
|
3 |
+
size 31344561
|