Fix the code

app.py

@@ -38,7 +38,7 @@ class GradioApp:
         )
 
     def launch(self):
-        with gr.Blocks() as interface:
+        with gr.Blocks(css="#small_video { width: 400px !important; height: 300px !important; }") as interface:
             # Input for the prompt at the top
             prompt_input = gr.Textbox(label="Input Prompt", elem_id="input_textbox")
 
@@ -50,12 +50,14 @@ class GradioApp:
                 with gr.Column():
                     image_output = gr.Image(label="Generated Image", elem_id="generated_image")
                 with gr.Column():
-                    video_output = gr.Video(label="3D Model Video", elem_id="model_video")
+                    video_output = gr.Video(label="3D Model Video", elem_id="small_video")
 
             # Generate the image first
             def generate_image_and_display(prompt):
+                modified_prompt = prompt + ", isolated, on a plain background, minimal, no extra objects"
+                print(modified_prompt)
                 # Generate the image from the prompt
-                image_data = self.sdxl_generator.generate_image([prompt])[0]
+                image_data = self.sdxl_generator.generate_image([modified_prompt])[0]
                 return Image.open(BytesIO(image_data))
 
             # Generate the 3D after the image is ready