Spaces:

matteopilotto
/

foodvision_mini

Runtime error

matteopilotto commited on Oct 29, 2022

Commit

16d81eb

1 Parent(s): c37b2cf

remove model.to('cpu')

Files changed (1) hide show

app.py CHANGED Viewed

@@ -18,7 +18,6 @@ def predict(img: PIL.Image) -> Tuple[Dict, float]:
     img = preprocess(img).unsqueeze(dim=0)
-    model.to('cpu')
     model.eval()
     with torch.inference_mode():
         probs = model(img).softmax(dim=-1).squeeze().tolist()

     img = preprocess(img).unsqueeze(dim=0)
     model.eval()
     with torch.inference_mode():
         probs = model(img).softmax(dim=-1).squeeze().tolist()