Update app.py

app.py

@@ -162,11 +162,59 @@ def generate_image(prompt,
         return [Image.fromarray(images[i]).resize((768, 768), Image.LANCZOS) for i in range(parallel_size)]
 
 
+
+# Custom CSS as a string
+custom_css = """
+    .gradio-container {
+        font-family: 'Inter', -apple-system, sans-serif;
+    }
+    .image-preview {
+        min-height: 300px;
+        max-height: 500px;
+        width: 100%;
+        object-fit: contain;
+        border-radius: 8px;
+        border: 2px solid #eee;
+    }
+    .tab-nav {
+        background: white;
+        padding: 1rem;
+        border-radius: 8px;
+        box-shadow: 0 2px 4px rgba(0,0,0,0.05);
+    }
+    .examples-table {
+        font-size: 0.9rem;
+    }
+    .gr-button.gr-button-lg {
+        padding: 12px 24px;
+        font-size: 1.1rem;
+    }
+    .gr-input, .gr-select {
+        border-radius: 6px;
+    }
+    .gr-form {
+        background: white;
+        padding: 20px;
+        border-radius: 12px;
+        box-shadow: 0 4px 6px rgba(0,0,0,0.05);
+    }
+    .gr-panel {
+        border: none;
+        background: transparent;
+    }
+    .footer {
+        text-align: center;
+        margin-top: 2rem;
+        padding: 1rem;
+        color: #666;
+    }
+"""
+
 # Gradio interface with improved UI
-with gr.Blocks(theme=gr.themes.Soft(
-    primary_hue="blue",
-    secondary_hue="indigo",
-)) as demo:
+with gr.Blocks(
+    theme=gr.themes.Soft(primary_hue="blue", secondary_hue="indigo"),
+    css=custom_css
+) as demo:
     gr.Markdown(
         """
         # Deepseek Multimodal
@@ -181,7 +229,7 @@ with gr.Blocks(theme=gr.themes.Soft(
     
     with gr.Tabs():
         # Visual Chat Tab
-        with gr.Tab("Visual Understanding"):  # Removed icon parameter
+        with gr.Tab("Visual Understanding"):
             with gr.Row(equal_height=True):
                 with gr.Column(scale=1):
                     image_input = gr.Image(
@@ -222,8 +270,7 @@ with gr.Blocks(theme=gr.themes.Soft(
                     
                     understanding_button = gr.Button(
                         "Analyze Image",
-                        variant="primary",
-                        size="lg"
+                        variant="primary"
                     )
             
             understanding_output = gr.Textbox(
@@ -244,7 +291,7 @@ with gr.Blocks(theme=gr.themes.Soft(
                 )
 
         # Image Generation Tab
-        with gr.Tab("Image Generation"):  # Removed icon parameter
+        with gr.Tab("Image Generation"):
             with gr.Column():
                 prompt_input = gr.Textbox(
                     label="Image Description",
@@ -278,8 +325,7 @@ with gr.Blocks(theme=gr.themes.Soft(
                 
                 generation_button = gr.Button(
                     "Generate Images",
-                    variant="primary",
-                    size="lg"
+                    variant="primary"
                 )
                 
                 image_output = gr.Gallery(
@@ -315,54 +361,6 @@ with gr.Blocks(theme=gr.themes.Soft(
         outputs=image_output
     )
 
-# Add custom CSS
-demo.load(css="""
-    .gradio-container {
-        font-family: 'Inter', -apple-system, sans-serif;
-    }
-    .image-preview {
-        min-height: 300px;
-        max-height: 500px;
-        width: 100%;
-        object-fit: contain;
-        border-radius: 8px;
-        border: 2px solid #eee;
-    }
-    .tab-nav {
-        background: white;
-        padding: 1rem;
-        border-radius: 8px;
-        box-shadow: 0 2px 4px rgba(0,0,0,0.05);
-    }
-    .examples-table {
-        font-size: 0.9rem;
-    }
-    .gr-button.gr-button-lg {
-        padding: 12px 24px;
-        font-size: 1.1rem;
-    }
-    .gr-input, .gr-select {
-        border-radius: 6px;
-    }
-    .gr-form {
-        background: white;
-        padding: 20px;
-        border-radius: 12px;
-        box-shadow: 0 4px 6px rgba(0,0,0,0.05);
-    }
-    .gr-panel {
-        border: none;
-        background: transparent;
-    }
-    .footer {
-        text-align: center;
-        margin-top: 2rem;
-        padding: 1rem;
-        color: #666;
-    }
-""")
-
 # Launch the demo
 if __name__ == "__main__":
     demo.launch(share=True)
-