Spaces:

Sayed223
/

ecg-ai-classifier

Sleeping

Sayed223 commited on 27 days ago

Commit

a97991a

verified ·

1 Parent(s): 4c92995

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -48,7 +48,7 @@ def generate_report(image):
         pred_index = torch.argmax(probabilities).item()
         pred_class = classes[pred_index]
-    # Prepare probability text
     prob_text = ""
     for i, cls in enumerate(classes):
         prob_text += f"{cls}: {probabilities[i] * 100:.2f}%\n"

         pred_index = torch.argmax(probabilities).item()
         pred_class = classes[pred_index]
+    # Prepare probability texts
     prob_text = ""
     for i, cls in enumerate(classes):
         prob_text += f"{cls}: {probabilities[i] * 100:.2f}%\n"