Update app.py

app.py

@@ -31,8 +31,8 @@ gradio_ui = gr.Interface(
         ["In the quiet hours, even the shadows seem too heavy to bear."],
         ["Riding a tempest of emotions, where ecstatic highs crash into desolate lows without warning."]
     ],
-    inputs=gr.inputs.Textbox(lines=10, label="Write the text here"),
-    outputs=gr.outputs.Label(num_top_classes=5, type="auto", label="Mental Health Disorder Category"),
+    inputs=gr.Textbox(lines=10, label="Write the text here"),
+    outputs=gr.Label(num_top_classes=4, type="auto", label="Mental Health Disorder Category"),
     theme="huggingface",
     article="<p style='text-align: center'>Please read the tutorial to fine-tune the Llama 3.1 model on Mental Health Classification <a href='https://www.datacamp.com/tutorial/fine-tuning-llama-3-1' target='_blank'>https://www.datacamp.com/tutorial/fine-tuning-llama-3-1</a></p>",
 )