Spaces:

remotewith
/

dented_final

Build error

remotewith commited on Jun 29, 2023

Commit

fa64015

1 Parent(s): 53b5a12

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -52,7 +52,7 @@ def predict(img):
     img1 = effnetb2_transforms(img1).unsqueeze(0)
     pix = normalize_2d(np.array(img))
-    pix1=model.predict(str(image), confidence=40, overlap=30).[0].cpu().numpy()
     # Put model into evaluation mode and turn on inference mode

     img1 = effnetb2_transforms(img1).unsqueeze(0)
     pix = normalize_2d(np.array(img))
+    pix1=model.predict(str(image), confidence=40, overlap=30).numpy()
     # Put model into evaluation mode and turn on inference mode