Update app.py

app.py

@@ -47,35 +47,24 @@ footer {display:none !important}
     background: none rgb(66, 133, 244) !important;
     box-shadow: rgb(0 0 0 / 23%) 0px 1px 7px 0px !important;
 }
-
-/* Two-column layout for text and labels */
-.gr-blocks-container {
-    display: flex;
-    gap: 20px;
-}
-.gr-blocks-container .gr-block:first-child {
-    flex: 1;
-}
-.gr-blocks-container .gr-block:last-child {
-    flex: 1;
-}
 """
 
-iface = gr.Interface(
-    fn=classify_text,
-    inputs=[
-        gr.Dropdown(AVAILABLE_MODELS, label="Choose Model"),
-        gr.Row([
-            gr.Textbox(label="Enter Text", placeholder="Type or paste text here..."),
-            gr.Textbox(label="Enter Labels (comma-separated)", placeholder="e.g., sports, politics, technology")
-        ])
-    ],
-    outputs=gr.Label(label="Classification Scores"),
-    title="Zero-Shot Text Classifier",
-    description="Select a model, enter text, and a set of labels to classify it using a zero-shot classification model.",
-    examples=examples,
-    css=css
-)
+with gr.Blocks(css=css) as iface:
+    gr.Markdown("# Zero-Shot Text Classifier")
+    gr.Markdown("Select a model, enter text, and a set of labels to classify it using a zero-shot classification model.")
+    
+    model_dropdown = gr.Dropdown(AVAILABLE_MODELS, label="Choose Model")
+    
+    with gr.Row():
+        text_input = gr.Textbox(label="Enter Text", placeholder="Type or paste text here...")
+        label_input = gr.Textbox(label="Enter Labels (comma-separated)", placeholder="e.g., sports, politics, technology")
+    
+    output_label = gr.Label(label="Classification Scores")
+    
+    submit_button = gr.Button("Classify")
+    submit_button.click(fn=classify_text, inputs=[model_dropdown, text_input, label_input], outputs=output_label)
+    
+    gr.Examples(examples, inputs=[text_input, label_input])
 
 # Launch the app
 if __name__ == "__main__":