Update app.py

app.py

@@ -129,10 +129,7 @@ def inference():
 prediction, targets = inference()
 prediction = np.array(prediction) >= 0.5
 
-print(prediction)
 st.write(prediction)
-
-print(targets)
 st.write(targets)
 
 targets = np.argmax(targets, axis=1)
@@ -141,8 +138,8 @@ accuracy = metrics.accuracy_score(targets, prediction)
 f1_score_micro = metrics.f1_score(targets, prediction, average="micro")
 f1_score_macro = metrics.f1_score(targets, prediction, average="macro")
 
-print(prediction)
 st.write(prediction)
+st.write(targets)
 st.write(f"Accuracy Score = {accuracy}")
 st.write(f"F1 Score (Micro) = {f1_score_micro}")
 st.write(f"F1 Score (Macro) = {f1_score_macro}")