logger test2

app.py

@@ -179,11 +179,14 @@ def api_classification(img, rank: int): #  -> dict[str, float]:
     for i in torch.nonzero(probs > min_prob).squeeze():
         output[" ".join(txt_names[i][0][: rank + 1])] += probs[i]
 
-    logger.info(">>>>")
-    logger.info(probs[0])
-    return {"message": probs[0]}
-    # topk_names = heapq.nlargest(k, output, key=output.get)
-    # return {name: output[name] for name in topk_names}
+    topk_names = heapq.nlargest(k, output, key=output.get)
+
+    exampleB = topk_names[0]
+    logger.info("exampleB")
+    logger.info(exampleB)
+    logger.info("output(exampleB)")
+    logger.info(output(exampleB))
+    return {name: output[name] for name in topk_names}
 
 
 def change_output(choice):