Update app.py
Browse files
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('
|
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 = []
|