Model loads to CPU now:

app.py

@@ -75,7 +75,7 @@ def load_model():
         url = 'https://drive.google.com/uc?id=1KO-QXUBfwzjauWLhiVi9StD3y0GtiBbj'
         gdown.download(url, model_path, quiet=False)
 
-    model = torch.load(model_path)
+    model = torch.load(model_path, map_location=torch.device('cpu'))
     model.to(device)
 
     return model
@@ -141,4 +141,4 @@ team in this task achieved an F1 score of around 70.
 We believe that the low overall scores are due to the dataset's tagging process, as it appeared inconsistent and at times inaccurate.
 """
 
-st.markdown(text, unsafe_allow_html=True)
+st.markdown(text, unsafe_allow_html=True)