Update app.py

app.py

@@ -96,34 +96,39 @@ def predict(content, policy):
     else:
         return f'VIOLATING ({decoded_output[-1]})'
 
-# Create Gradio interface
-iface = gr.Interface(
-    fn=predict,
-    inputs=[gr.Textbox(label="Content", lines=2, value=DEFAULT_CONTENT), 
-            gr.Textbox(label="Policy", lines=10, value=DEFAULT_POLICY)], 
-    outputs="label",
-    title="CoPE Alpha Preview",
-    description="See if the given content violates your given policy.",
-    article="""
+
+with gr.Blocks() as iface:
+    gr.Markdown("# CoPE Alpha Preview")
+    gr.Markdown("See if the given content violates your given policy.")
+    
+    with gr.Row():
+        content_input = gr.Textbox(label="Content", lines=2, value=DEFAULT_CONTENT)
+        policy_input = gr.Textbox(label="Policy", lines=10, value=DEFAULT_POLICY)
+    
+    submit_btn = gr.Button("Submit")
+    output = gr.Label(label="Label")
+    
+    gr.Markdown("""
     ## About CoPE
     
-    CoPE (the COntent Policy Evaluation engine) is a small language model capable of accurate content policy labeling. This is a *preview* of our alpha release and is strictly for *research* purposes. This should *NOT* be used for any production use cases.
+    CoPE (the COntent Policy Evaluation engine) is a small language model capable of accurate content policy labeling. This is a **preview** of our alpha release and is strictly for **research** purposes. This should **NOT** be used for any production use cases.
     
-    ### How to Use
+    ## How to Use
                                                                                                                                                                                                                                                                        
     1. Enter your content in the "Content" box.
     2. Specify your policy in the "Policy" box.
     3. Click "Submit" to see the results.
 
-    *Note*: Inference times are *very slow* (30-45 seconds) since this is built on dev infra and not yet optimized for live systems. Please be patient while testing!
+    **Note**: Inference times are **very slow** (30-45 seconds) since this is built on dev infra and not yet optimized for live systems. Please be patient while testing!
     
-    ### Feedback
+    ## More Info
 
     - [Give us feedback](https://example.com) to help us improve
     - Read our FAQ to learn more about CoPE
     - Join our mailing list to keep in touch
-    """
-)
+    """)
+    
+    submit_btn.click(predict, inputs=[content_input, policy_input], outputs=output)
 
 # Launch the app
 iface.launch()