Update app.py

app.py

@@ -5,6 +5,11 @@ from datasets import load_dataset
 from sklearn.model_selection import train_test_split
 import subprocess
 
+from sklearn.metrics import classification_report
+
+
+
+
 # Function to show value counts of 'Hair Loss'
 def show_hair_loss_counts():
     return df['Hair Loss'].value_counts().to_json()  # Convert to JSON for display
@@ -112,5 +117,12 @@ with gr.Blocks() as demo:
     run_button = gr.Button("Run Command")
     run_button.click(run_command, inputs=command_input, outputs=command_output)
 
+
+
+
+y_pred = model.predict(X_test)
+print(classification_report(y_test, y_pred))
+
+
 demo.launch()