fix

app.py

@@ -12,7 +12,7 @@ def predict(image):
 
     model = EfficientNet.from_name('efficientnet-b7', num_classes=2)
     model_weights_path = 'efficientnetb7_tyrequality_classifier.pth'
-    model.load_state_dict(torch.load(model_weights_path))
+    model.load_state_dict(torch.load(model_weights_path, map_location=torch.device('cpu')))
     model.eval()
 
     transform = transforms.Compose([
@@ -34,7 +34,7 @@ def predict(image):
     _, predicted_class = torch.max(output, 1)
 
     probs = torch.nn.functional.softmax(output, dim=1)
-    print(probs, "probs")
+    # print(probs, "probs")
     conf, _ = torch.max(probs, 1)
 
     result = "Tire status is {} with confidence level in {}%".format(class_to_label[predicted_class.item()], conf.item()*100)