MeYourHint commited on
Commit
980e069
Β·
1 Parent(s): 340bd0a
Files changed (1) hide show
  1. app.py +10 -10
app.py CHANGED
@@ -34,8 +34,7 @@ from gen_t2m import load_vq_model, load_res_model, load_trans_model, load_len_es
34
 
35
  clip_version = 'ViT-B/32'
36
 
37
- Total_Calls = 7792
38
- WEBSITE = f"""
39
  <div class="embed_hidden">
40
  <h1 style='text-align: center'> MoMask: Generative Masked Modeling of 3D Human Motions </h1>
41
  <h2 style='text-align: center'>
@@ -55,7 +54,6 @@ WEBSITE = f"""
55
  <a target="_blank" href="https://ericguo5513.github.io/source_files/momask_2023_bib.txt"> <button type="button" class="btn btn-primary btn-lg"> BibTex </button></a> &ensp;
56
  <a target="_blank" href="https://huggingface.co/spaces/MeYourHint/MoMask?docker=true" title="Run with docker"> <button type="button" class="btn btn-primary btn-lg"> Docker </button></a> &ensp;
57
  <a target="_blank" href="https://huggingface.co/spaces/MeYourHint/MoMask?duplicate=true" title="Duplicate space with private gpu and no queue"> <button type="button" class="btn btn-primary btn-lg"> Duplicate </button></a> &ensp;
58
- <img src=https://img.shields.io/badge/Visit-{Total_Calls}-green>
59
  </h3>
60
  <h3> Description </h3>
61
  <p>
@@ -65,7 +63,7 @@ WEBSITE = f"""
65
  πŸš€πŸš€πŸš€ In addition, we provide a link to download the generated human skeletal motion in <b>BVH</b> file format, compatible with CG software such as Blender!!! πŸš€πŸš€πŸš€
66
  </p>
67
  <p>
68
- 😁😁😁 If you find this demo interesting, we would appreciate your star <a href="https://github.com/EricGuo5513/momask-codes" target="_blank"><img alt="GitHub Repo stars" src="https://img.shields.io/github/stars/EricGuo5513/momask-codes"></a>. More details could be found on our <a href='https://ericguo5513.github.io/momask/' target='_blank'>webpage</a>. 🫢🫢🫢
69
  </p>
70
  <p>
71
  If you have any issues on this space or feature requests, we warmly welcome you to contact us through our <a href="https://github.com/EricGuo5513/momask-codes/issues" target="_blank">github repo</a> or <a href="mailto:ymu3@ualberta.ca?subject =[MoMask]Feedback&body = Message">email</a>.
@@ -73,6 +71,7 @@ If you have any issues on this space or feature requests, we warmly welcome you
73
  </div>
74
  """
75
  WEBSITE_bottom = """
 
76
  <p>
77
  We thanks <a href="https://huggingface.co/spaces/Mathux/TMR" target="_blank">TMR</a> for this cool space template.
78
  </p>
@@ -129,11 +128,12 @@ if not os.path.exists("/data/stats"):
129
  with open("/data/stats/Prompts.text", 'w') as f:
130
  pass
131
 
132
- def update_total_calls():
133
- global Total_Calls
134
- Total_Calls_offset = 7792 ## init number from visit, 01/07
135
- with open("/data/stats/Prompts.text", 'r') as f:
136
- Total_Calls = len(f.readlines()) + Total_Calls_offset
 
137
 
138
  ### Load Stats ###
139
 
@@ -204,7 +204,7 @@ def generate(
204
  print(text)
205
  with open("/data/stats/Prompts.text", 'a') as f:
206
  f.write(text+'\n')
207
- update_total_calls()
208
  prompt_list = []
209
  length_list = []
210
  est_length = False
 
34
 
35
  clip_version = 'ViT-B/32'
36
 
37
+ WEBSITE = """
 
38
  <div class="embed_hidden">
39
  <h1 style='text-align: center'> MoMask: Generative Masked Modeling of 3D Human Motions </h1>
40
  <h2 style='text-align: center'>
 
54
  <a target="_blank" href="https://ericguo5513.github.io/source_files/momask_2023_bib.txt"> <button type="button" class="btn btn-primary btn-lg"> BibTex </button></a> &ensp;
55
  <a target="_blank" href="https://huggingface.co/spaces/MeYourHint/MoMask?docker=true" title="Run with docker"> <button type="button" class="btn btn-primary btn-lg"> Docker </button></a> &ensp;
56
  <a target="_blank" href="https://huggingface.co/spaces/MeYourHint/MoMask?duplicate=true" title="Duplicate space with private gpu and no queue"> <button type="button" class="btn btn-primary btn-lg"> Duplicate </button></a> &ensp;
 
57
  </h3>
58
  <h3> Description </h3>
59
  <p>
 
63
  πŸš€πŸš€πŸš€ In addition, we provide a link to download the generated human skeletal motion in <b>BVH</b> file format, compatible with CG software such as Blender!!! πŸš€πŸš€πŸš€
64
  </p>
65
  <p>
66
+ 😁😁😁 If you find this demo interesting, we would appreciate your star on our <a href="https://github.com/EricGuo5513/momask-codes" target="_blank">github repo</a>. More details could be found on our <a href='https://ericguo5513.github.io/momask/' target='_blank'>webpage</a>. 🫢🫢🫢
67
  </p>
68
  <p>
69
  If you have any issues on this space or feature requests, we warmly welcome you to contact us through our <a href="https://github.com/EricGuo5513/momask-codes/issues" target="_blank">github repo</a> or <a href="mailto:ymu3@ualberta.ca?subject =[MoMask]Feedback&body = Message">email</a>.
 
71
  </div>
72
  """
73
  WEBSITE_bottom = """
74
+ <div class="embed_hidden">
75
  <p>
76
  We thanks <a href="https://huggingface.co/spaces/Mathux/TMR" target="_blank">TMR</a> for this cool space template.
77
  </p>
 
128
  with open("/data/stats/Prompts.text", 'w') as f:
129
  pass
130
 
131
+ # Total_Calls = 7792
132
+ # def update_total_calls():
133
+ # global Total_Calls
134
+ # Total_Calls_offset = 7792 ## init number from visit, 01/07
135
+ # with open("/data/stats/Prompts.text", 'r') as f:
136
+ # Total_Calls = len(f.readlines()) + Total_Calls_offset
137
 
138
  ### Load Stats ###
139
 
 
204
  print(text)
205
  with open("/data/stats/Prompts.text", 'a') as f:
206
  f.write(text+'\n')
207
+ # update_total_calls()
208
  prompt_list = []
209
  length_list = []
210
  est_length = False