Upload infer.sh with huggingface_hub

infer.sh

@@ -7,8 +7,6 @@ MAX_LENGTH=4096
 TOP_K=1
 OUTPUT_IMAGE_PATH="./results/"
 FORMAT_KEY="status_action_op_sensitive"
-INPUT_JSON="/inspire/hdd/ws-ba572160-47f8-4ca1-984e-d6bcdeb95dbb/a100-maybe/albus/ICCV_2025/CogAgent/dataset/"
-OUTPUT_JSON="/inspire/hdd/ws-ba572160-47f8-4ca1-984e-d6bcdeb95dbb/a100-maybe/albus/ICCV_2025/CogAgent/results/json/"
 
 # 定义执行次数
 NUM_RUNS=170  # 设置执行次数
@@ -16,6 +14,8 @@ NUM_RUNS=170  # 设置执行次数
 # 循环执行 Python 脚本多次
 for i in $(seq 1 $NUM_RUNS)
 do
+  INPUT_JSON="/inspire/hdd/ws-ba572160-47f8-4ca1-984e-d6bcdeb95dbb/a100-maybe/albus/ICCV_2025/CogAgent/dataset/$i.json"
+  OUTPUT_JSON="/inspire/hdd/ws-ba572160-47f8-4ca1-984e-d6bcdeb95dbb/a100-maybe/albus/ICCV_2025/CogAgent/results/json/$i.json"
   echo "Running script, round $i..."
   python cogagent_infer.py \
     --model_dir "$MODEL_DIR" \
@@ -24,6 +24,6 @@ do
     --top_k $TOP_K \
     --output_image_path f"$OUTPUT_IMAGE_PATH{i}" \
     --format_key "$FORMAT_KEY" \
-    --input_json "$INPUT_JSON{i}.json" \
-    --output_json "$OUTPUT_JSON{i}.json"
+    --input_json "$INPUT_JSON" \
+    --output_json "$OUTPUT_JSON"
 done