FIX: Simplified probability mapping in Gradio UI to resolve 100% display issue.

app.py

@@ -97,23 +97,25 @@ async def get_predict(text: str):
         raise HTTPException(status_code=500, detail=f"Internal Server Error during GET prediction: {e}")
 
 def gradio_interface_fn(text_input):
+    """Interface function to be called by Gradio UI."""
     if not text_input or not text_input.strip():
         return "Please enter text for classification.", None
         
     try:
         result = predict_svm([text_input])[0]
+        
         prediction_label = result.prediction
         
         prob_display = {
-            result.prediction: result.probability_STOP if result.prediction == LABEL_1 else result.probability_NOT_STOP,
-            LABEL_1 if result.prediction == LABEL_0 else LABEL_0: result.probability_STOP if result.prediction == LABEL_0 else result.probability_NOT_STOP
+            LABEL_0: result.probability_NOT_STOP,  
+            LABEL_1: result.probability_STOP       
         }
         
         return prediction_label, prob_display
         
     except Exception as e:
         return f"An error occurred: {str(e)}", None
-
+    
 ui = gr.Interface(
     fn=gradio_interface_fn,
     inputs=gr.Textbox(lines=2, placeholder="Enter a message to classify...", label="Input Text"),