Update app.py

app.py

@@ -125,21 +125,19 @@ def inference():
             final_outputs.extend(torch.sigmoid(outputs).cpu().detach().numpy().tolist())
     return final_outputs, final_targets
 
-
 prediction, targets = inference()
-#prediction = np.array(prediction) >= 0.5
-
+st.write("before argmax")
 st.write(prediction)
 st.write(targets)
 
+st.write("after argmax axis 1")
 targets = np.argmax(targets, axis=1)
 prediction = np.argmax(prediction, axis=1)
-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")
+st.write(prediction)
+st.write(targets)
 
+st.write("after argmax axis 0")
+targets = np.argmax(targets, axis=0)
+prediction = np.argmax(prediction, axis=0)
 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}")