Spaces:
Runtime error
Runtime error
mayuema
commited on
Commit
•
0ff3cdf
1
Parent(s):
d06f4b9
first release
Browse files- app.py +0 -1
- inference_followyourpose.py +3 -0
app.py
CHANGED
@@ -11,7 +11,6 @@ from inference_followyourpose import merge_config_then_run
|
|
11 |
|
12 |
HF_TOKEN = os.getenv('HF_TOKEN')
|
13 |
pipe = merge_config_then_run()
|
14 |
-
pipe.download_model()
|
15 |
|
16 |
with gr.Blocks(css='style.css') as demo:
|
17 |
gr.HTML(
|
|
|
11 |
|
12 |
HF_TOKEN = os.getenv('HF_TOKEN')
|
13 |
pipe = merge_config_then_run()
|
|
|
14 |
|
15 |
with gr.Blocks(css='style.css') as demo:
|
16 |
gr.HTML(
|
inference_followyourpose.py
CHANGED
@@ -5,6 +5,8 @@ import copy
|
|
5 |
import gradio as gr
|
6 |
from transformers import AutoTokenizer, CLIPTextModel
|
7 |
from huggingface_hub import snapshot_download
|
|
|
|
|
8 |
|
9 |
def get_time_string() -> str:
|
10 |
x = datetime.datetime.now()
|
@@ -18,6 +20,7 @@ class merge_config_then_run():
|
|
18 |
self.text_encoder = None
|
19 |
self.vae = None
|
20 |
self.unet = None
|
|
|
21 |
|
22 |
def download_model(self):
|
23 |
REPO_ID = 'YueMafighting/FollowYourPose_v1'
|
|
|
5 |
import gradio as gr
|
6 |
from transformers import AutoTokenizer, CLIPTextModel
|
7 |
from huggingface_hub import snapshot_download
|
8 |
+
import sys
|
9 |
+
sys.path.append('FollowYourPose')
|
10 |
|
11 |
def get_time_string() -> str:
|
12 |
x = datetime.datetime.now()
|
|
|
20 |
self.text_encoder = None
|
21 |
self.vae = None
|
22 |
self.unet = None
|
23 |
+
self.download_model()
|
24 |
|
25 |
def download_model(self):
|
26 |
REPO_ID = 'YueMafighting/FollowYourPose_v1'
|