Update app.py

app.py

@@ -36,8 +36,11 @@ def predict_news_type(content):
     probabilities = content_predict[0].tolist()
     category_probabilities = {label_dict[str(i)]: prob for i, prob in enumerate(probabilities)}
     
-    # Convert the dictionary to a list of strings
-    category_probabilities_list = [f"{category}: {probability * 100:.2f}%" for category, probability in category_probabilities.items()]
+    # Sort the category probabilities in descending order
+    sorted_category_probabilities = sorted(category_probabilities.items(), key=lambda x: x[1], reverse=True)
+    
+    # Convert the sorted dictionary to a list of strings
+    category_probabilities_list = [f"{category}: {probability * 100:.2f}%" for category, probability in sorted_category_probabilities]
     
     # Convert the list to a single string
     category_probabilities_str = "\n".join(category_probabilities_list)
@@ -58,4 +61,4 @@ demo = gr.Interface(
 
 if __name__ == "__main__":
     # Launch the Gradio Interface
-    demo.launch()  # server_name=host, server_port=port, share=True
+    demo.launch()