Spaces:

muellerzr
/

deployment-no-fastai

Runtime error

muellerzr HF Staff commited on Feb 22, 2023

Commit

8a128ee

1 Parent(s): 0c3d87d

Rm cuda

Files changed (1) hide show

src/app.py CHANGED Viewed

@@ -27,7 +27,6 @@ vocab = [
 model = get_model()
 state = torch.load('exported_model.pth')["model"]
 apply_weights(model, state, copy_weight)
-model.cuda()
 to_tensor = ToTensor()
 norm = Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225])
@@ -37,7 +36,7 @@ def classify_image(inp):
     transformed_input = pad(crop(inp, (460, 460)), (460, 460))
     transformed_input = to_tensor(transformed_input).unsqueeze(0)
     transformed_input = gpu_crop(transformed_input, (224, 224))
-    transformed_input = norm(transformed_input).cuda()
     model.eval()
     with torch.no_grad():
         pred = model(transformed_input)

 model = get_model()
 state = torch.load('exported_model.pth')["model"]
 apply_weights(model, state, copy_weight)
 to_tensor = ToTensor()
 norm = Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225])
     transformed_input = pad(crop(inp, (460, 460)), (460, 460))
     transformed_input = to_tensor(transformed_input).unsqueeze(0)
     transformed_input = gpu_crop(transformed_input, (224, 224))
+    transformed_input = norm(transformed_input)
     model.eval()
     with torch.no_grad():
         pred = model(transformed_input)