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 = []
|