Spaces:

Wootang01
/

image_classifier_four

Runtime error

Wootang01 commited on Apr 7, 2022

Commit

1fccc85

1 Parent(s): ad55483

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -14,7 +14,7 @@ transform = create_transform(
 model.eval()
-def predict_fn(img)"
   img = img.convert('RGB')
   img = transform(img).unsqueeze(0)
   with torch.no_grad():

 model.eval()
+def predict_fn(img):
   img = img.convert('RGB')
   img = transform(img).unsqueeze(0)
   with torch.no_grad():