Spaces:
Running
on
T4
Running
on
T4
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")
|