Spaces:
Sleeping
Sleeping
Commit
·
d4580e4
1
Parent(s):
4d2bb34
Update app.py
Browse files
app.py
CHANGED
|
@@ -6,8 +6,7 @@ model = torch.hub.load(
|
|
| 6 |
"generator",
|
| 7 |
pretrained=True)
|
| 8 |
face2paint = torch.hub.load(
|
| 9 |
-
'AK391/animegan2-pytorch:main', 'face2paint'
|
| 10 |
-
size=512)
|
| 11 |
def inference(img):
|
| 12 |
out = face2paint(model, img)
|
| 13 |
return out
|
|
|
|
| 6 |
"generator",
|
| 7 |
pretrained=True)
|
| 8 |
face2paint = torch.hub.load(
|
| 9 |
+
'AK391/animegan2-pytorch:main', 'face2paint')
|
|
|
|
| 10 |
def inference(img):
|
| 11 |
out = face2paint(model, img)
|
| 12 |
return out
|