Spaces:
Running
on
Zero
Running
on
Zero
Commit
·
98bef81
1
Parent(s):
7757ac2
revert to normal selection
Browse files- model/model_manager.py +2 -2
model/model_manager.py
CHANGED
|
@@ -195,12 +195,12 @@ class ModelManager:
|
|
| 195 |
def generate_video_vg_museum_parallel_anony(self, model_A, model_B):
|
| 196 |
# Using list comprehension to get the difference between two lists
|
| 197 |
picking_list = [item for item in self.model_vg_list if item not in self.excluding_model_list]
|
| 198 |
-
picking_list = [item for item in picking_list if item not in self.desired_model_list]
|
| 199 |
if model_A == "" and model_B == "":
|
| 200 |
model_names = random.sample([model for model in picking_list], 2)
|
| 201 |
|
| 202 |
#override the random selection
|
| 203 |
-
model_names[random.choice([0, 1])] = random.choice(self.desired_model_list)
|
| 204 |
else:
|
| 205 |
model_names = [model_A, model_B]
|
| 206 |
|
|
|
|
| 195 |
def generate_video_vg_museum_parallel_anony(self, model_A, model_B):
|
| 196 |
# Using list comprehension to get the difference between two lists
|
| 197 |
picking_list = [item for item in self.model_vg_list if item not in self.excluding_model_list]
|
| 198 |
+
#picking_list = [item for item in picking_list if item not in self.desired_model_list]
|
| 199 |
if model_A == "" and model_B == "":
|
| 200 |
model_names = random.sample([model for model in picking_list], 2)
|
| 201 |
|
| 202 |
#override the random selection
|
| 203 |
+
#model_names[random.choice([0, 1])] = random.choice(self.desired_model_list)
|
| 204 |
else:
|
| 205 |
model_names = [model_A, model_B]
|
| 206 |
|