Only show image and video

app.py

@@ -36,19 +36,18 @@ class GradioApp:
 
     def full_pipeline(self, prompt):
         initial_image = self.sdxl_generator.generate_images([prompt])[0]
-        # controlled_image = self.controlnet_processor.controlnet_image(initial_image)
         mesh_data, video_data = self.video_generator.generate_3d_video(initial_image)
         
         # Create temporary files to display mesh and video content
-        with tempfile.NamedTemporaryFile(delete=False, suffix=".obj") as mesh_file:
-            mesh_file.write(mesh_data)
-            mesh_path = mesh_file.name
+        #with tempfile.NamedTemporaryFile(delete=False, suffix=".obj") as mesh_file:
+            #mesh_file.write(mesh_data)
+            #mesh_path = mesh_file.name
         
         with tempfile.NamedTemporaryFile(delete=False, suffix=".mp4") as video_file:
             video_file.write(video_data)
             video_path = video_file.name
         
-        return initial_image, mesh_path, video_path
+        return initial_image, video_path
 
     def launch(self):
         interface = gr.Interface(
@@ -56,13 +55,13 @@ class GradioApp:
             inputs=gr.Textbox(label="Input Prompt"),
             outputs=[
                 gr.Image(label="Generated Image"),
-                gr.File(label="3D Mesh (.obj)"),
+                #gr.File(label="3D Mesh (.obj)"),
                 gr.Video(label="3D Model Video")
             ],
             title="SDXL to ControlNet to 3D Pipeline",
             description="Generate an image using SDXL, refine it with ControlNet, and generate a 3D video output."
         )
-        interface.launch(share=True)  # Added `share=True` for public link
+        interface.launch(share=True)
 
 if __name__ == "__main__":
     app = GradioApp()