Ahsen Khaliq commited on
Commit
b767bd0
1 Parent(s): ac969df

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -6,7 +6,7 @@ model2 = torch.hub.load(
6
  "generator",
7
  pretrained=True, # or give URL to a pretrained model
8
  device="cuda", # or "cpu" if you don't have a GPU
9
- progress=False, # show progress
10
  )
11
 
12
  model1 = torch.hub.load("bryandlee/animegan2-pytorch:main", "generator", pretrained="face_paint_512_v1", device="cuda")
6
  "generator",
7
  pretrained=True, # or give URL to a pretrained model
8
  device="cuda", # or "cpu" if you don't have a GPU
9
+ progress=False
10
  )
11
 
12
  model1 = torch.hub.load("bryandlee/animegan2-pytorch:main", "generator", pretrained="face_paint_512_v1", device="cuda")