BeanSamuel commited on
Commit
cc28da2
·
1 Parent(s): 79889bc
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("./generator_epoch_70.pth", map_location=device))
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:7622d51270be899f993f9999e7f402c63cf6dfbdb6c37525c3bc8ddd67294b02
3
- size 31344538
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:926b6b12ad992cb2b4b1fc835f08d7f6c7ec881d76598eca80d592081790b553
3
+ size 31344561