Spaces:

ruescog
/

practica3

Runtime error

Rubén Escobedo commited on Mar 19, 2022

Commit

6c792e5

1 Parent(s): da9c964

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -25,6 +25,8 @@ def predict(img):
     with torch.no_grad():
         outputs = model(tensor)
     mask = np.array(outputs.cpu())
     mask[mask == 1] = 255 # grape
     mask[mask == 2] = 150 # leaves

     with torch.no_grad():
         outputs = model(tensor)
+    outputs = torch.argmax(outputs,1)
     mask = np.array(outputs.cpu())
     mask[mask == 1] = 255 # grape
     mask[mask == 2] = 150 # leaves