Spaces:
Sleeping
Sleeping
ionosphere
commited on
Commit
•
17a695b
1
Parent(s):
c4bef56
Fix bug
Browse files- rag_module.py +4 -4
rag_module.py
CHANGED
@@ -15,10 +15,6 @@ from langchain_mistralai.chat_models import ChatMistralAI
|
|
15 |
#chain modules
|
16 |
from langchain.chains import RetrievalQA
|
17 |
|
18 |
-
# login for accessing repo model
|
19 |
-
from huggingface_hub import login
|
20 |
-
login(token=os.environ.get("HFTOKEN"))
|
21 |
-
|
22 |
# import PyPDF2
|
23 |
import os
|
24 |
import re
|
@@ -26,6 +22,10 @@ from dotenv import load_dotenv
|
|
26 |
load_dotenv()
|
27 |
from collections import defaultdict
|
28 |
|
|
|
|
|
|
|
|
|
29 |
|
30 |
class RagModule():
|
31 |
def __init__(self):
|
|
|
15 |
#chain modules
|
16 |
from langchain.chains import RetrievalQA
|
17 |
|
|
|
|
|
|
|
|
|
18 |
# import PyPDF2
|
19 |
import os
|
20 |
import re
|
|
|
22 |
load_dotenv()
|
23 |
from collections import defaultdict
|
24 |
|
25 |
+
# login for accessing repo model
|
26 |
+
from huggingface_hub import login
|
27 |
+
login(token=os.environ.get("HFTOKEN"))
|
28 |
+
|
29 |
|
30 |
class RagModule():
|
31 |
def __init__(self):
|