use gpu if available

similarity_fast.py

@@ -19,7 +19,8 @@ category_pickle_file_path = f'./embeddings/fast/{filename}-categories.pkl'
 class SimilarityFast:
     def __init__(self, db_cursor):
         self.db_cursor = db_cursor
-        self.model = SentenceTransformer(model_name)
+        self.device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+        self.model = SentenceTransformer(model_name).to(self.device)
         
         self.db_cursor.execute("SELECT description FROM dictionary")
         dictionary = self.db_cursor.fetchall()