Spaces:
Running
Running
Update app.py
Browse files
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/
|
| 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/
|
| 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/
|
| 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/
|
| 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/
|
| 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/
|
| 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")
|