Update app.py

app.py

@@ -127,7 +127,7 @@ def process_input(text_input, labels_or_premise, mode):
         # Global prediction
         global_pred = nli_classifier([{"text": text_input, "text_pair": labels_or_premise}], return_all_scores=True)[0]
         global_results = {pred['label']: pred['score'] for pred in global_pred}
-        global_label = max(global_results.items(), key=lambda x: x[1])[0]
+        global_label, global_confidence = max(global_results.items(), key=lambda x: x[1])
         
         # Sentence-level analysis
         sentences = sent_tokenize(text_input)
@@ -143,7 +143,7 @@ def process_input(text_input, labels_or_premise, mode):
                 'scores': sent_scores
             })
         
-        analysis_html = create_analysis_html(sentence_results, global_label)
+        analysis_html = create_analysis_html(sentence_results, global_label,global_confidence)
         return global_results, analysis_html
 
 def update_interface(mode):
@@ -168,7 +168,7 @@ def update_interface(mode):
     else:  # Long Context NLI
         return (
             gr.update(
-                label="🔎 Global Hypothesis",
+                label="🔎 Hypothesis",
                 placeholder="Enter a hypothesis to test against the full context...",
                 value=long_context_examples[0][1]
             ),