reqs, app

app.py

@@ -38,7 +38,7 @@ def inference(img_dir):
    image = np.array(Image.open(str(img_dir)).convert('L'))
    image = transforms(image=image)['image']
    with torch.inference_mode():
-        out = model(image).item()
+        out = model(image.unsqueeze(0)).item()
    out = out > 0
    return "Fracture +" if out else "Fracture -"
 

requirements.txt

@@ -1,2 +1,3 @@
 albumentations
-torch
+torch
+transformers