Update application

app.py

@@ -21,7 +21,13 @@ def predict(system_prompt, user_prompt, selected_model):
         predicted_label, probabilities = cross_encoder_predict_relevance(system_prompt, user_prompt, ms_model, ms_tokenizer, device)
 
     probability_off_topic = probabilities[0][1] * 100
-    result = f'{probability_off_topic:.3f}% chance this is off-topic'
+    label = "Off-topic" if predicted_label==1 else "On-topic"
+    result = f"""
+    **Prediction Summary**:
+
+    - **Predicted Label**: {label}
+    - **Probability of Off-topic**: {probability_off_topic:.3f}%
+    """
 
     return result
 
@@ -43,7 +49,7 @@ with gr.Blocks(theme=gr.themes.Soft(), fill_height=True) as app:
     # Button to run the prediction
     get_classfication = gr.Button("Check Content")
 
-    output_result = gr.Textbox(label="Classification and Probabilities", lines=5)
+    output_result = gr.Markdown(label="Classification and Probabilities")
 
     get_classfication.click(
         fn=predict,