Spaces:
Runtime error
Runtime error
Ahsen Khaliq
commited on
Commit
•
2697d0c
1
Parent(s):
334026c
Update app.py
Browse files
app.py
CHANGED
@@ -152,8 +152,9 @@ ckptspider = torch.load('Spiderverse-face-500iters-8face.pt', map_location=lambd
|
|
152 |
generatorspider.load_state_dict(ckptspider["g"], strict=False)
|
153 |
|
154 |
|
155 |
-
def inference(img, model):
|
156 |
-
|
|
|
157 |
|
158 |
my_w = projection(aligned_face, "test.pt", device).unsqueeze(0)
|
159 |
if model == 'JoJo':
|
|
|
152 |
generatorspider.load_state_dict(ckptspider["g"], strict=False)
|
153 |
|
154 |
|
155 |
+
def inference(img, model):
|
156 |
+
img.save('out.jpg')
|
157 |
+
aligned_face = align_face('out.jpg')
|
158 |
|
159 |
my_w = projection(aligned_face, "test.pt", device).unsqueeze(0)
|
160 |
if model == 'JoJo':
|