amosyou commited on
Commit
53b5c44
1 Parent(s): f643383

clean up app

Browse files
Files changed (1) hide show
  1. app.py +2 -3
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
- hf_hub_download(repo_id=REPO_ID, filename=FILENAME)
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()