Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -1,6 +1,9 @@
|
|
1 |
import spaces
|
2 |
import gradio as gr
|
3 |
|
|
|
|
|
|
|
4 |
# Define repository and local directory
|
5 |
repo_id = "ai-forever/GHOST-2.0-repo" # HF repo
|
6 |
local_dir = "./" # Target local directory
|
@@ -29,9 +32,6 @@ from src.utils.preblending import calc_pseudo_target_bg
|
|
29 |
from train_aligner import AlignerModule
|
30 |
from train_blender import BlenderModule
|
31 |
|
32 |
-
from huggingface_hub import snapshot_download
|
33 |
-
|
34 |
-
|
35 |
|
36 |
@spaces.GPU
|
37 |
def infer_headswap(source, target):
|
|
|
1 |
import spaces
|
2 |
import gradio as gr
|
3 |
|
4 |
+
|
5 |
+
from huggingface_hub import snapshot_download
|
6 |
+
|
7 |
# Define repository and local directory
|
8 |
repo_id = "ai-forever/GHOST-2.0-repo" # HF repo
|
9 |
local_dir = "./" # Target local directory
|
|
|
32 |
from train_aligner import AlignerModule
|
33 |
from train_blender import BlenderModule
|
34 |
|
|
|
|
|
|
|
35 |
|
36 |
@spaces.GPU
|
37 |
def infer_headswap(source, target):
|