Spaces:
Paused
Paused
Ahsen Khaliq
commited on
Commit
·
b767bd0
1
Parent(s):
ac969df
Update app.py
Browse files
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
|
| 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")
|