Update app.py

app.py

@@ -32,8 +32,9 @@ def process_input(text_input, labels_or_premise, mode):
         results = {label: score for label, score in zip(prediction['labels'], prediction['scores'])}
         return results, ''
     else:  # NLI mode
-        pred = nli_classifier([{"text": text_input, "text_pair": labels_or_premise}], return_all_scores=True)[0]
-        results = {pred['label']: pred['score'] for pred in pred}
+        pred= nli_classifier([{"text": text_input, "text_pair": labels_or_premise}],return_all_scores=True)[0]
+        results= {pred['label']:pred['score'] for pred in pred}
+
         return results, ''
 
 def update_interface(mode):
@@ -41,17 +42,19 @@ def update_interface(mode):
         return (
             gr.update(
                 label="🏷️ Categories", 
-                placeholder="Enter comma-separated categories..."
+                placeholder="Enter comma-separated categories...",
+                value=zero_shot_examples[0][1]
             ),
-            gr.update()
+            gr.update(value=zero_shot_examples[0][0])
         )
     else:
         return (
             gr.update(
                 label="🔎 Hypothesis", 
-                placeholder="Enter a hypothesis to compare with the premise..."
+                placeholder="Enter a hypothesis to compare with the premise...",
+                value=nli_examples[0][1]
             ),
-            gr.update()
+            gr.update(value=nli_examples[0][0])
         )
 
 with gr.Blocks() as demo:
@@ -75,14 +78,14 @@ with gr.Blocks() as demo:
             label="✍️ Input Text",
             placeholder="Enter your text...",
             lines=3,
-            value=zero_shot_examples[0][0]
+            value=zero_shot_examples[0][0]  # Initial value
         )
         
         labels_or_premise = gr.Textbox(
             label="🏷️ Categories",
             placeholder="Enter comma-separated categories...",
             lines=2,
-            value=zero_shot_examples[0][1]
+            value=zero_shot_examples[0][1]  # Initial value
         )
         
         submit_btn = gr.Button("Submit")
@@ -92,15 +95,11 @@ with gr.Blocks() as demo:
             gr.Markdown(label="📈 Analysis", visible=False)
         ]
 
-        # Create a dummy textbox to trigger updates when examples are selected
-        example_trigger = gr.Textbox(visible=False)
-
         with gr.Column(variant="panel") as zero_shot_examples_panel:
             gr.Examples(
                 examples=zero_shot_examples,
                 inputs=[text_input, labels_or_premise],
                 label="Zero-Shot Classification Examples",
-                outputs=example_trigger
             )
     
         with gr.Column(variant="panel") as nli_examples_panel:
@@ -108,8 +107,7 @@ with gr.Blocks() as demo:
                 examples=nli_examples,
                 inputs=[text_input, labels_or_premise],
                 label="Natural Language Inference Examples",
-                outputs=example_trigger
-            )
+        )
 
     def update_visibility(mode):
         return (
@@ -117,7 +115,6 @@ with gr.Blocks() as demo:
             gr.update(visible=(mode == "Natural Language Inference"))
         )
 
-    # Update interface on mode change
     mode.change(
         fn=update_interface,
         inputs=[mode],
@@ -130,13 +127,6 @@ with gr.Blocks() as demo:
         outputs=[zero_shot_examples_panel, nli_examples_panel]
     )
     
-    # Update the label whenever an example is selected
-    example_trigger.change(
-        fn=update_interface,
-        inputs=[mode],
-        outputs=[labels_or_premise, text_input]
-    )
-    
     submit_btn.click(
         fn=process_input,
         inputs=[text_input, labels_or_premise, mode],