Spaces:
Running
Running
gavinyuan
commited on
Commit
•
31c3841
1
Parent(s):
c0e9547
update: app.py
Browse files
app.py
CHANGED
|
@@ -69,7 +69,7 @@ def extract_generator(ckpt: str, pt: str):
|
|
| 69 |
|
| 70 |
''' load model '''
|
| 71 |
if fs_model_name == 'faceshifter':
|
| 72 |
-
pt_path = make_abs_path("./weights/extracted/
|
| 73 |
# pt_path = make_abs_path("../ffplus/extracted_ckpt/G_mouth1_t512_6.pth")
|
| 74 |
# ckpt_path = "/apdcephfs/share_1290939/gavinyuan/out/triplet512_6/epoch=3-step=128999.ckpt"
|
| 75 |
# pt_path = make_abs_path("../ffplus/extracted_ckpt/G_mouth1_t512_4.pth")
|
|
|
|
| 69 |
|
| 70 |
''' load model '''
|
| 71 |
if fs_model_name == 'faceshifter':
|
| 72 |
+
pt_path = make_abs_path("./weights/extracted/G_mouth1_t38_post.pth")
|
| 73 |
# pt_path = make_abs_path("../ffplus/extracted_ckpt/G_mouth1_t512_6.pth")
|
| 74 |
# ckpt_path = "/apdcephfs/share_1290939/gavinyuan/out/triplet512_6/epoch=3-step=128999.ckpt"
|
| 75 |
# pt_path = make_abs_path("../ffplus/extracted_ckpt/G_mouth1_t512_4.pth")
|