Update app.py

app.py

@@ -129,7 +129,10 @@ 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)
@@ -138,6 +141,7 @@ 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(f"Accuracy Score = {accuracy}")
 st.write(f"F1 Score (Micro) = {f1_score_micro}")