Spaces:

Robb49
/

Image_Origin_Classification

Sleeping

Robb49 commited on Jul 4

Commit

94777e9

•

1 Parent(s): dc5dd91

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -113,7 +113,7 @@ chance_dict = {}
 def one_prediction(img):
     preds = {classname: 0 for classname in classes}
-    img = transform(img).to(device)
     img.unsqueeze_(0)
     models = [dense_net, googlenet, vgg13, res_net, model_5]
     with torch.no_grad():

 def one_prediction(img):
     preds = {classname: 0 for classname in classes}
+    img = transform(img)
     img.unsqueeze_(0)
     models = [dense_net, googlenet, vgg13, res_net, model_5]
     with torch.no_grad():