Spaces:

jerilseb
/

quickdraw-small

Sleeping

jerilseb commited on Aug 24

Commit

6ab4125

•

1 Parent(s): 8958d40

update stuff

Files changed (1) hide show

app.py CHANGED Viewed

@@ -40,7 +40,7 @@ transform = transforms.Compose([
 def predict(image):
     image = image['composite']
-    image = Image.fromarray(image).convert('RGB')
     input = transform(image).unsqueeze(0)
     print(input.shape)

 def predict(image):
     image = image['composite']
+    image = Image.fromarray(image).convert('L')
     input = transform(image).unsqueeze(0)
     print(input.shape)