Add cpu support

app.py

@@ -14,6 +14,10 @@ from streamlit_plotly_events import plotly_events
 import plotly.graph_objects as go
 import logging
 import utils
+
+use_cpu = torch.cuda.is_available()
+device = "cpu" if use_cpu else "cuda"
+
 # Activate tqdm with pandas
 tqdm.pandas()
 
@@ -54,7 +58,7 @@ df = load_data()
 # Caching the model and tokenizer to avoid reloading
 @st.cache_resource
 def load_model_and_tokenizer():
-    encoder = AutoModel.from_pretrained("sentence-transformers/gtr-t5-base").encoder.to("cuda")
+    encoder = AutoModel.from_pretrained("sentence-transformers/gtr-t5-base").encoder.to(device)
     tokenizer = AutoTokenizer.from_pretrained("sentence-transformers/gtr-t5-base")
     return encoder, tokenizer