Spaces:
Running
on
T4
Running
on
T4
Ahsen Khaliq
commited on
Commit
·
9ad2cb7
1
Parent(s):
306b921
Update app.py
Browse files
app.py
CHANGED
|
@@ -18,7 +18,7 @@ face2paint = torch.hub.load(
|
|
| 18 |
)
|
| 19 |
|
| 20 |
def inference(img, ver):
|
| 21 |
-
if ver == 'version 2':
|
| 22 |
out = face2paint(model2, img)
|
| 23 |
else:
|
| 24 |
out = face2paint(model1, img)
|
|
|
|
| 18 |
)
|
| 19 |
|
| 20 |
def inference(img, ver):
|
| 21 |
+
if ver == 'version 2 (🔺 robustness,🔻 stylization)':
|
| 22 |
out = face2paint(model2, img)
|
| 23 |
else:
|
| 24 |
out = face2paint(model1, img)
|