Spaces:

jiang20
/

test

Runtime error

jiang20 commited on Mar 20, 2023

Commit

7e89ea8

1 Parent(s): 26e2141

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -35,6 +35,7 @@ def greet_backdoor(image):
         return 'changed'
     else:
         image = transform_nor(image).unsqueeze(0)
         output = model(image).squeeze()
         return 'classified: ' + torch.argmax(output)

         return 'changed'
     else:
         image = transform_nor(image).unsqueeze(0)
+        print(image.shape)
         output = model(image).squeeze()
         return 'classified: ' + torch.argmax(output)