Update genshin_impact_couple_app.py

genshin_impact_couple_app.py

@@ -12,6 +12,13 @@ import random
 # Fix the random seed for reproducibility
 random.seed(42)  # You can use any fixed number as the seed
 
+# Define common combinations
+common_combinations = [
+    ("lookaround", "back_and_forth"),  # Most common
+    ("lookdown", "back_and_forth"),    # Less common
+    ("rotate360", "headbanging")       # Rare
+]
+
 # 下载数据集
 repo_id = "svjack/3DitScene_cache"
 folder_path = "Genshin-Impact-Couple-with-Tags-IID-Gender-Only-Two-Joy-Caption_Head10"
@@ -34,15 +41,11 @@ for example in dataset:
         'joy-caption': example['joy_caption_surrounding']
     })
 
-# 定义可用的 gen_camerapath 和 render_camerapath 选项
-gen_camerapath_options = ['lookaround', 'lookdown', 'rotate360']
-render_camerapath_options = ['back_and_forth', 'llff', 'headbanging']
-
 # 为每个例子随机分配 gen_camerapath 和 render_camerapath
 examples_with_combinations = []
 for example in examples:
-    gen_camerapath = random.choice(gen_camerapath_options)
-    render_camerapath = random.choice(render_camerapath_options)
+    # Randomly select a combination from common_combinations
+    gen_camerapath, render_camerapath = random.choice(common_combinations)
     examples_with_combinations.append({
         'image': example['image'],
         'joy-caption': example['joy-caption'],