Spaces:
Runtime error
Runtime error
Commit
•
b4cd79a
1
Parent(s):
5308e54
Update main.py
Browse files
main.py
CHANGED
@@ -15,6 +15,8 @@ from langchain.vectorstores import Chroma
|
|
15 |
from langchain.vectorstores import FAISS
|
16 |
from transformers import AutoTokenizer, TextStreamer, pipeline
|
17 |
|
|
|
|
|
18 |
DEVICE = "cuda:0" if torch.cuda.is_available() else "cpu"
|
19 |
|
20 |
|
|
|
15 |
from langchain.vectorstores import FAISS
|
16 |
from transformers import AutoTokenizer, TextStreamer, pipeline
|
17 |
|
18 |
+
"""Patient Response : hi doctor \n\n"What brings you here today? Please provide me with some background information about your health so far." \nPatient Response : im arafath i have fever"""
|
19 |
+
|
20 |
DEVICE = "cuda:0" if torch.cuda.is_available() else "cpu"
|
21 |
|
22 |
|