Spaces:
Runtime error
Runtime error
Ahsen Khaliq
commited on
Commit
•
ffd64f5
1
Parent(s):
7d9079d
Update app.py
Browse files
app.py
CHANGED
|
@@ -6,7 +6,7 @@ model = torch.hub.load("bryandlee/animegan2-pytorch:main", "generator", pretrain
|
|
| 6 |
|
| 7 |
face2paint = torch.hub.load(
|
| 8 |
'bryandlee/animegan2-pytorch:main', 'face2paint',
|
| 9 |
-
size=512, device="
|
| 10 |
)
|
| 11 |
def inference(img):
|
| 12 |
out = face2paint(model, img)
|
|
|
|
| 6 |
|
| 7 |
face2paint = torch.hub.load(
|
| 8 |
'bryandlee/animegan2-pytorch:main', 'face2paint',
|
| 9 |
+
size=512, device="cpu"
|
| 10 |
)
|
| 11 |
def inference(img):
|
| 12 |
out = face2paint(model, img)
|