Update app.py

app.py

@@ -34,7 +34,7 @@ def load_model():
                                                         num_labels=8,
                                                         output_attentions=False,
                                                         output_hidden_states=False)
-    model.load_state_dict(torch.load(CUSTOMMODEL_PATH))
+    model.load_state_dict(torch.load(CUSTOMMODEL_PATH, map_location ='cpu'))
     return model, tokenizer