maze commited on
Commit
34bfe14
β€’
1 Parent(s): 9107020

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -1,7 +1,7 @@
1
  from huggingface_hub import hf_hub_download
2
 
3
 
4
- Rain_Princess = hf_hub_download(repo_id="maze/FastStyleTransfer", filename="Rain_Princess.pth")
5
  #modelarcanev3 = hf_hub_download(repo_id="akhaliq/ArcaneGANv0.3", filename="ArcaneGANv0.3.jit")
6
  #modelarcanev2 = hf_hub_download(repo_id="akhaliq/ArcaneGANv0.2", filename="ArcaneGANv0.2.jit")
7
 
@@ -112,7 +112,7 @@ transformer = TransformerNetwork().to(device)
112
  transformer.eval()
113
 
114
  transform = transforms.Compose([
115
- transforms.Resize(256),
116
  transforms.ToTensor(),
117
  transforms.Normalize(mean, std),
118
  ])
 
1
  from huggingface_hub import hf_hub_download
2
 
3
 
4
+ Rain_Princess = hf_hub_download(repo_id="maze/FastStyleTransfer", filename="Rain_Princess_512.pth")
5
  #modelarcanev3 = hf_hub_download(repo_id="akhaliq/ArcaneGANv0.3", filename="ArcaneGANv0.3.jit")
6
  #modelarcanev2 = hf_hub_download(repo_id="akhaliq/ArcaneGANv0.2", filename="ArcaneGANv0.2.jit")
7
 
 
112
  transformer.eval()
113
 
114
  transform = transforms.Compose([
115
+ transforms.Resize(512),
116
  transforms.ToTensor(),
117
  transforms.Normalize(mean, std),
118
  ])