mayuema commited on
Commit
0ff3cdf
1 Parent(s): d06f4b9

first release

Browse files
Files changed (2) hide show
  1. app.py +0 -1
  2. 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'