Spaces:
Sleeping
Sleeping
Commit
·
aeb5e49
1
Parent(s):
47587a5
Update app.py
Browse files
app.py
CHANGED
|
@@ -40,7 +40,7 @@ PROMPT = PromptTemplate(
|
|
| 40 |
)
|
| 41 |
chain_type_kwargs = {"prompt": PROMPT}
|
| 42 |
|
| 43 |
-
llm = HuggingFaceHub(repo_id="
|
| 44 |
|
| 45 |
qachain=RetrievalQA.from_chain_type(llm, retriever=vectorstore.as_retriever(), chain_type_kwargs=chain_type_kwargs)
|
| 46 |
st.header("#CodeWars localGPT", divider='rainbow')
|
|
|
|
| 40 |
)
|
| 41 |
chain_type_kwargs = {"prompt": PROMPT}
|
| 42 |
|
| 43 |
+
llm = HuggingFaceHub(repo_id="google/flan-ul2", model_kwargs={"temperature":0.1, "max_new_tokens":150})
|
| 44 |
|
| 45 |
qachain=RetrievalQA.from_chain_type(llm, retriever=vectorstore.as_retriever(), chain_type_kwargs=chain_type_kwargs)
|
| 46 |
st.header("#CodeWars localGPT", divider='rainbow')
|