Update app.py

app.py

@@ -35,8 +35,12 @@ def classify_comment(comment):
     # Make predictions
     predictions = model.predict(comment_sequence)[0]
     results = dict(zip(labels, predictions))
+    
+    max_value = max(results.values())
 
-    return results
+    max_keys = [key for key, value in results.items() if value == max_value]
+
+    return max_keys[0].capitalize()
 
 # Create the Gradio interface
 comment_input = gr.inputs.Textbox(label="Enter your comment here")