Spaces:

hafez1082
/

skinwise-pytorch

Sleeping

hafez1082 commited on May 29

Commit

84061a8

verified ·

1 Parent(s): 4f83924

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -57,8 +57,9 @@ async def predict_skin_disease(file: UploadFile = File(...)):
         confidence = probabilities[0][predicted_class_idx].item() * 100
         return JSONResponse(content={
-            "predicted_class": predicted_class_name,
-            "confidence": float(confidence),
         })
     except Exception as e:

         confidence = probabilities[0][predicted_class_idx].item() * 100
         return JSONResponse(content={
+            "model": "Pytorch",
+            "class": predicted_class_name,
+            "probability": f"{confidence:.2f}%",
         })
     except Exception as e: