Spaces:
Runtime error
Runtime error
Commit
•
a5ce063
1
Parent(s):
34de110
add cuda
Browse files
app.py
CHANGED
|
@@ -9,6 +9,8 @@ from haystack.nodes import BM25Retriever, PromptTemplate, AnswerParser, PromptNo
|
|
| 9 |
import os
|
| 10 |
from haystack.pipelines import Pipeline
|
| 11 |
|
|
|
|
|
|
|
| 12 |
# Load environment variables (if any)
|
| 13 |
load_dotenv(".env")
|
| 14 |
load_dotenv()
|
|
|
|
| 9 |
import os
|
| 10 |
from haystack.pipelines import Pipeline
|
| 11 |
|
| 12 |
+
os.environ["CUDA_VISIBLE_DEVICES"] = "-1"
|
| 13 |
+
|
| 14 |
# Load environment variables (if any)
|
| 15 |
load_dotenv(".env")
|
| 16 |
load_dotenv()
|