ymcmy commited on
Commit
1e41eb8
·
verified ·
1 Parent(s): 41384a0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -215,7 +215,7 @@ class TextureContrastClassifier(nn.Module):
215
 
216
  input_shape = (128, 256)
217
  model = TextureContrastClassifier(input_shape)
218
- model.load_state_dict(torch.load('C:/Users/Matt/Downloads/model_epoch_45.pth', map_location=torch.device('cpu')))
219
 
220
  def inference(image, model):
221
  predictions = []
 
215
 
216
  input_shape = (128, 256)
217
  model = TextureContrastClassifier(input_shape)
218
+ model.load_state_dict(torch.load('./model_epoch_36.pth', map_location=torch.device('cpu')))
219
 
220
  def inference(image, model):
221
  predictions = []