Spaces:

for876543
/

plant-id-3

Build error

for876543 commited on Oct 14, 2022

Commit

feab5d7

•

1 Parent(s): 82de39e

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -21,7 +21,7 @@ def classify_image(inp):
   inp = Image.fromarray(inp)
   inp= Resize((300, 300))(inp)
   inp= ToTensor()(inp)
-  inp.append(torch.unsqueeze(inp, 0))
   ##print(inp.shape)
   #inp = inp.astype(np.uint8).reshape((-1, 3, 300, 300))
   ##print(inp.shape)

   inp = Image.fromarray(inp)
   inp= Resize((300, 300))(inp)
   inp= ToTensor()(inp)
+  inp = torch.unsqueeze(inp, 0)
   ##print(inp.shape)
   #inp = inp.astype(np.uint8).reshape((-1, 3, 300, 300))
   ##print(inp.shape)