YANF

app.py

@@ -68,7 +68,7 @@ def predict_image(img):
     )
 
     result = classification_conformalizer.conformalize(inference_pred)
-    list_of_classes = [dataset.idx_to_cls[i] for i in result.detach().numpy()[0]]
+    list_of_classes = [dataset.idx_to_cls[i] for i in result[0].detach().numpy()]
     result = f"Predicted classes with 90% confidence: {list_of_classes}"
     return img_old, result