Upload app.py

app.py

@@ -224,7 +224,8 @@ def generate_question_for_scene_dict(scene):
 def generate_counterfactual_scenes(num_scenes, num_objects, min_objects, max_objects, num_counterfactuals, 
                                     cf_types, same_cf_type, min_change_score, max_cf_attempts, min_noise_level,
                                     output_dir, blender_path=None, use_gpu=0, samples=512, 
-                                    width=320, height=240, skip_render=False, generate_questions=False):
+                                    width=320, height=240, skip_render=False, generate_questions=False,
+                                    semantic_only=False, negative_only=False):
     if not PIPELINE_AVAILABLE:
         return {
             'success': False,
@@ -864,7 +865,9 @@ def main():
                     width=image_width,
                     height=image_height,
                     skip_render=False,
-                    generate_questions=generate_questions
+                    generate_questions=generate_questions,
+                    semantic_only=semantic_only,
+                    negative_only=negative_only
                 )
                 
                 progress_bar.progress(80)