Cuda support?

app.py

@@ -16,10 +16,7 @@ from nltk.tokenize import sent_tokenize
 import nltk
 nltk.download('punkt')
 
-
-if not torch.cuda.is_available():
-    print("Warning: No GPU found. Please add GPU to your notebook")
-
+device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
 import pandas as pd
 st.title('Sociology Paragraph Search')
@@ -82,7 +79,7 @@ def search(query, top_k=40):
 
     ##### Sematic Search #####
     # Encode the query using the bi-encoder and find potentially relevant passages
-    question_embedding = bi_encoder.encode(query, convert_to_tensor=True)
+    question_embedding = bi_encoder.encode(query, convert_to_tensor=True).to(device)
 
 
     hits = util.semantic_search(question_embedding, corpus_embeddings, top_k=top_k)