tirendazakademi commited on
Commit
f72e0ab
1 Parent(s): 2f49426

modified app file

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -27,7 +27,7 @@ class ImageClassifier(nn.Module):
27
  return output
28
 
29
  model = ImageClassifier()
30
- model.load_state_dict(torch.load('comet-torch-model.pth'))
31
 
32
  def predict(inp):
33
  image_transform = transforms.Compose([ transforms.Resize(size=(224,224)), transforms.ToTensor()])
 
27
  return output
28
 
29
  model = ImageClassifier()
30
+ model.load_state_dict(torch.load('model-data_comet-torch-model'))
31
 
32
  def predict(inp):
33
  image_transform = transforms.Compose([ transforms.Resize(size=(224,224)), transforms.ToTensor()])