Update main.py

main.py

@@ -35,6 +35,7 @@ async def get_prediction(file: UploadFile = File(...)):
     for r in results:
         print(names[int(r.probs.top1)]) 
         class_name = names[int(r.probs.top1)]
-        confidence = r.probs.top1conf
+        confidence = str(r.probs.top1conf)
+        print(str(r.probs.top1conf))
 
     return {"class_name":class_name,"confidence":confidence}