app.py

app.py

@@ -36,11 +36,11 @@ model.eval()
 
 def inference(img_dir):
    image = np.array(Image.open(str(img_dir)).convert('L'))
-    image = transforms(image=image)['image']
-    with torch.inference_mode():
+   image = transforms(image=image)['image']
+   with torch.inference_mode():
         out = model(image).item()
-    out = out > 0
-    return "Fracture +" if out else "Fracture -"
+   out = out > 0
+   return "Fracture +" if out else "Fracture -"
 
 examples = ["NF1.jpg", "NF2.jpg", "NF3.jpg", "F1.jpg", "F2.jpg", "F3.jpg"]