modified app file

app.py

@@ -27,7 +27,7 @@ class ImageClassifier(nn.Module):
         return output
         
 model = ImageClassifier()
-model.load_state_dict(torch.load('model-data_comet-torch-model'))
+model.load_state_dict(torch.load('model-data_comet-torch-model.pth'))
 
 def predict(inp):
     image_transform = transforms.Compose([ transforms.Resize(size=(224,224)), transforms.ToTensor()])