Spaces:
Running
on
T4
Running
on
T4
Ahsen Khaliq
commited on
Commit
•
9a9f589
1
Parent(s):
50b7dca
Update app.py
Browse files
app.py
CHANGED
|
@@ -10,7 +10,7 @@ model2 = torch.hub.load(
|
|
| 10 |
progress=False, # show progress
|
| 11 |
)
|
| 12 |
|
| 13 |
-
model1 = torch.hub.load("bryandlee/animegan2-pytorch:main", "generator", pretrained="face_paint_512_v1")
|
| 14 |
|
| 15 |
|
| 16 |
face2paint = torch.hub.load(
|
|
|
|
| 10 |
progress=False, # show progress
|
| 11 |
)
|
| 12 |
|
| 13 |
+
model1 = torch.hub.load("bryandlee/animegan2-pytorch:main", "generator", pretrained="face_paint_512_v1", device="cuda")
|
| 14 |
|
| 15 |
|
| 16 |
face2paint = torch.hub.load(
|