Update StreamlitModel.py

StreamlitModel.py

@@ -20,8 +20,8 @@ import torch.hub
 profanity.load_censor_words()    
 
 #It seems the streamlit space does not allow mps
-device = 'cuda' if torch.cuda.is_available() else 'mps' if torch.backends.mps.is_built() else 'cpu'
-#device = 'cuda' if torch.cuda.is_available() else 'cpu'
+#device = 'cuda' if torch.cuda.is_available() else 'mps' if torch.backends.mps.is_built() else 'cpu'
+device = 'cuda' if torch.cuda.is_available() else 'cpu'
     
 class GPT2_Model(GPT2PreTrainedModel):