Update generate_3d_object.py

generate_3d_object.py

@@ -1,5 +1,6 @@
 import trimesh
 import gradio as gr
+import os
 
 def generate_3d_object(shape_type, radius, width, height):
     if shape_type == "Sphere":
@@ -26,7 +27,8 @@ iface = gr.Interface(
     ],
     outputs=gr.Model3D(label="Generated 3D Object"),
     title="3D Object Generator",
-    description="Generate 3D objects like spheres and boxes."
+    description="Generate 3D objects like spheres and boxes.",
+    flagging_dir=os.path.join(os.getcwd(), "flagged")  # Specify the flagging directory
 )
 
 # Launch the interface