Spaces:
Running
on
T4
Running
on
T4
Ahsen Khaliq
commited on
Commit
•
0d960e2
1
Parent(s):
89f4ad8
Update app.py
Browse files
app.py
CHANGED
@@ -14,7 +14,7 @@ model1 = torch.hub.load("bryandlee/animegan2-pytorch:main", "generator", pretrai
|
|
14 |
|
15 |
face2paint = torch.hub.load(
|
16 |
'bryandlee/animegan2-pytorch:main', 'face2paint',
|
17 |
-
size=512, device="cuda"
|
18 |
)
|
19 |
|
20 |
def inference(img, ver):
|
|
|
14 |
|
15 |
face2paint = torch.hub.load(
|
16 |
'bryandlee/animegan2-pytorch:main', 'face2paint',
|
17 |
+
size=512, device="cuda"
|
18 |
)
|
19 |
|
20 |
def inference(img, ver):
|