ynhe commited on
Commit
7fa242c
·
verified ·
1 Parent(s): 6d6a708

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -7
app.py CHANGED
@@ -34,6 +34,7 @@ for i in hf_api.list_repo_tree('Vchitect/VBench-2.0_sampled_videos',repo_type='d
34
  model_name = i.path
35
  if '.git' not in model_name and '.md' not in model_name:
36
  model_names.append(model_name)
 
37
 
38
  with open("videos_by_dimension.json") as f:
39
  dimension = json.load(f)['videos_by_dimension']
@@ -131,7 +132,7 @@ def display_videos(type, prompt, model_name_1, model_name_2):
131
  def record_user_feedback_a(model_name1, model_name2, type, prompt):
132
  # 0 means model A better, 1 means model B better, -1 means tie;
133
  hf_api.hf_hub_download(
134
- repo_id = "Vchitect/VBench_human_annotation",
135
  filename = "VBench2_arena_feedback.csv",
136
  repo_type = "dataset",
137
  local_dir = './'
@@ -141,7 +142,7 @@ def record_user_feedback_a(model_name1, model_name2, type, prompt):
141
  hf_api.upload_file(
142
  path_or_fileobj="VBench2_arena_feedback.csv",
143
  path_in_repo="VBench2_arena_feedback.csv",
144
- repo_id="Vchitect/VBench_human_annotation",
145
  token=hf_token,
146
  repo_type="dataset",
147
  commit_message="[From VBench Arena] user feedback",
@@ -151,7 +152,7 @@ def record_user_feedback_a(model_name1, model_name2, type, prompt):
151
  def record_user_feedback_b(model_name1, model_name2, type, prompt):
152
  # 0 means model A better, 1 means model B better , -1 means tie;
153
  hf_api.hf_hub_download(
154
- repo_id = "Vchitect/VBench_human_annotation",
155
  filename = "VBench2_arena_feedback.csv",
156
  repo_type = "dataset",
157
  local_dir = './'
@@ -161,7 +162,7 @@ def record_user_feedback_b(model_name1, model_name2, type, prompt):
161
  hf_api.upload_file(
162
  path_or_fileobj="VBench2_arena_feedback.csv",
163
  path_in_repo="VBench2_arena_feedback.csv",
164
- repo_id="Vchitect/VBench_human_annotation",
165
  token=hf_token,
166
  repo_type="dataset",
167
  commit_message="[From VBench Arena] user feedback",
@@ -171,7 +172,7 @@ def record_user_feedback_b(model_name1, model_name2, type, prompt):
171
  def record_user_feedback_tie(model_name1, model_name2, type, prompt):
172
  # 0 means model A better, 1 means model B better , -1 means tie;
173
  hf_api.hf_hub_download(
174
- repo_id = "Vchitect/VBench_human_annotation",
175
  filename = "VBench2_arena_feedback.csv",
176
  repo_type = "dataset",
177
  local_dir = './'
@@ -181,7 +182,7 @@ def record_user_feedback_tie(model_name1, model_name2, type, prompt):
181
  hf_api.upload_file(
182
  path_or_fileobj="VBench2_arena_feedback.csv",
183
  path_in_repo="VBench2_arena_feedback.csv",
184
- repo_id="Vchitect/VBench_human_annotation",
185
  token=hf_token,
186
  repo_type="dataset",
187
  commit_message="[From VBench Arena] user feedback",
@@ -256,4 +257,4 @@ with gr.Blocks() as interface:
256
  outputs=[feed0, feedt, feed1]
257
  )
258
 
259
- interface.launch()
 
34
  model_name = i.path
35
  if '.git' not in model_name and '.md' not in model_name:
36
  model_names.append(model_name)
37
+
38
 
39
  with open("videos_by_dimension.json") as f:
40
  dimension = json.load(f)['videos_by_dimension']
 
132
  def record_user_feedback_a(model_name1, model_name2, type, prompt):
133
  # 0 means model A better, 1 means model B better, -1 means tie;
134
  hf_api.hf_hub_download(
135
+ repo_id = "Vchitect/VBench-2.0_human_annotation",
136
  filename = "VBench2_arena_feedback.csv",
137
  repo_type = "dataset",
138
  local_dir = './'
 
142
  hf_api.upload_file(
143
  path_or_fileobj="VBench2_arena_feedback.csv",
144
  path_in_repo="VBench2_arena_feedback.csv",
145
+ repo_id="Vchitect/VBench-2.0_human_annotation",
146
  token=hf_token,
147
  repo_type="dataset",
148
  commit_message="[From VBench Arena] user feedback",
 
152
  def record_user_feedback_b(model_name1, model_name2, type, prompt):
153
  # 0 means model A better, 1 means model B better , -1 means tie;
154
  hf_api.hf_hub_download(
155
+ repo_id = "Vchitect/VBench-2.0_human_annotation",
156
  filename = "VBench2_arena_feedback.csv",
157
  repo_type = "dataset",
158
  local_dir = './'
 
162
  hf_api.upload_file(
163
  path_or_fileobj="VBench2_arena_feedback.csv",
164
  path_in_repo="VBench2_arena_feedback.csv",
165
+ repo_id="Vchitect/VBench-2.0_human_annotation",
166
  token=hf_token,
167
  repo_type="dataset",
168
  commit_message="[From VBench Arena] user feedback",
 
172
  def record_user_feedback_tie(model_name1, model_name2, type, prompt):
173
  # 0 means model A better, 1 means model B better , -1 means tie;
174
  hf_api.hf_hub_download(
175
+ repo_id = "Vchitect/VBench-2.0_human_annotation",
176
  filename = "VBench2_arena_feedback.csv",
177
  repo_type = "dataset",
178
  local_dir = './'
 
182
  hf_api.upload_file(
183
  path_or_fileobj="VBench2_arena_feedback.csv",
184
  path_in_repo="VBench2_arena_feedback.csv",
185
+ repo_id="Vchitect/VBench-2.0_human_annotation",
186
  token=hf_token,
187
  repo_type="dataset",
188
  commit_message="[From VBench Arena] user feedback",
 
257
  outputs=[feed0, feedt, feed1]
258
  )
259
 
260
+ interface.launch(server_name="0.0.0.0")