cuda bug

app_implementation.py

@@ -16,9 +16,17 @@ from search_utils import (
 
 
 class RetrievalApp:
+    def is_cuda_available(self):
+        try:
+            t = torch.Tensor([1]).cuda()
+        except:
+            return False
+        finally:
+            return True
+
     def get_device_options(self):
-        if torch.cuda.is_available:
-            return ["cuda", "cpu"]
+        if self.is_cuda_available():
+            return ["cpu", "cuda"]
         else:
             return ["cpu"]
 
@@ -67,9 +75,9 @@ class RetrievalApp:
                 st.table(get_repos_with_descriptions(retrieval_pipe.X_df, task_repos))
         with st.spinner(text="fetching results"):
             st.write(
-                retrieval_pipe.search(query, k, description_length, additional_shown_cols).to_html(
-                    escape=False, index=False
-                ),
+                retrieval_pipe.search(
+                    query, k, description_length, additional_shown_cols
+                ).to_html(escape=False, index=False),
                 unsafe_allow_html=True,
             )
         print("finished retrieval")