Spaces:
Running
on
Zero
Running
on
Zero
vinesmsuic
commited on
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 |
|