s

Deployment/app.py

@@ -171,7 +171,7 @@ if st.session_state['Loaded'] == False:
     # Offensiveness detection model 
 
     offensive_model = BertClassifier()
-    offensive_model.load_state_dict(torch.load(os.path.join(parent_path,'models/modelv3.pt')))
+    offensive_model.load_state_dict(torch.load(os.path.join(parent_path,'models/modelv3.pt'), map_location=torch.device('cpu')))
     offensive_tokenizer = BertTokenizer.from_pretrained('aubmindlab/bert-base-arabertv02', do_lower_case=True)
 
     #send model to device
@@ -193,7 +193,7 @@ if st.session_state['Loaded'] == False:
     racism_model,religionhate_model,verbalabuse_model,misogyny_model = MediumBert(),MediumBert(),MediumBert(),MediumBert()
     ################################################################
 
-    racism_model.load_state_dict(torch.load(os.path.join(parent_path,'models/racism/racism_arabert.pt')))
+    racism_model.load_state_dict(torch.load(os.path.join(parent_path,'models/racism/racism_arabert.pt'), map_location=torch.device('cpu')))
     racism_dict = {0: 'non_racist', 1: 'racist'}
 
     racism_model = racism_model.to(device)
@@ -204,7 +204,7 @@ if st.session_state['Loaded'] == False:
     print('Racism model loaded')
     ################################################################
 
-    religionhate_model.load_state_dict(torch.load(os.path.join(parent_path,'models/religion_hate/religion_hate_params.pt')))
+    religionhate_model.load_state_dict(torch.load(os.path.join(parent_path,'models/religion_hate/religion_hate_params.pt'), map_location=torch.device('cpu')))
     religionhate_dict = {0: 'Religion Hate', 1: 'Not Religion Hate'}
 
     religionhate_model = religionhate_model.to(device)
@@ -215,7 +215,7 @@ if st.session_state['Loaded'] == False:
     print('Religion Hate model loaded')
     ################################################################
 
-    verbalabuse_model.load_state_dict(torch.load(os.path.join(parent_path,'models/verbal_abuse/verbal_abuse_arabert.pt')))
+    verbalabuse_model.load_state_dict(torch.load(os.path.join(parent_path,'models/verbal_abuse/verbal_abuse_arabert.pt'), map_location=torch.device('cpu')))
     verbalabuse_dict = {0: 'Verbal Abuse', 1: 'Not Verbal Abuse'}
 
     verbalabuse_model=verbalabuse_model.to(device)
@@ -226,7 +226,7 @@ if st.session_state['Loaded'] == False:
     print('Verbal Abuse model loaded')
     ################################################################
 
-    misogyny_model.load_state_dict(torch.load(os.path.join(parent_path,'models/misogyny/misogyny.pt')))
+    misogyny_model.load_state_dict(torch.load(os.path.join(parent_path,'models/misogyny/misogyny.pt'), map_location=torch.device('cpu')))
     misogyny_dict = {0: 'misogyny', 1: 'non_misogyny'}
 
     misogyny_model=misogyny_model.to(device)
@@ -241,7 +241,7 @@ if st.session_state['Loaded'] == False:
     # Dialect detection model
 
     dialect_model = Dialect_Detection(10)
-    dialect_model.load_state_dict(torch.load(os.path.join(parent_path,'models/dialect_classifier.pt')))
+    dialect_model.load_state_dict(torch.load(os.path.join(parent_path,'models/dialect_classifier.pt'), map_location=torch.device('cpu')))
 
     dialect_model = dialect_model.to(device)