Spaces:
Sleeping
Sleeping
Update QnA.py
Browse files
QnA.py
CHANGED
@@ -67,7 +67,7 @@ def Q_A(vectorstore,question,API_KEY):
|
|
67 |
os.environ["GROQ_API_KEY"] = API_KEY
|
68 |
chat_llm = ChatGroq(model="llama3-8b-8192")
|
69 |
elif API_KEY.startswith('hf'):
|
70 |
-
chat_llm = get_hugging_face_model(
|
71 |
|
72 |
# Create a retriever
|
73 |
retriever = vectorstore.as_retriever(search_type = 'similarity',search_kwargs = {'k':2},)
|
|
|
67 |
os.environ["GROQ_API_KEY"] = API_KEY
|
68 |
chat_llm = ChatGroq(model="llama3-8b-8192")
|
69 |
elif API_KEY.startswith('hf'):
|
70 |
+
chat_llm = get_hugging_face_model(api_key=API_KEY)
|
71 |
|
72 |
# Create a retriever
|
73 |
retriever = vectorstore.as_retriever(search_type = 'similarity',search_kwargs = {'k':2},)
|