Update app.py
Browse files
app.py
CHANGED
|
@@ -41,7 +41,7 @@ if user_api_key is not None and user_api_key.strip() != "":
|
|
| 41 |
template=template
|
| 42 |
)
|
| 43 |
llm_chain = LLMChain(
|
| 44 |
-
llm=HuggingFaceHub(huggingfacehub_api_token=
|
| 45 |
prompt=prompt,
|
| 46 |
verbose=True,
|
| 47 |
memory=ConversationBufferWindowMemory(k=2)
|
|
|
|
| 41 |
template=template
|
| 42 |
)
|
| 43 |
llm_chain = LLMChain(
|
| 44 |
+
llm=HuggingFaceHub(huggingfacehub_api_token=user_api_key, repo_id="tiiuae/falcon-7b-instruct", model_kwargs={"temperature": 0.2}),
|
| 45 |
prompt=prompt,
|
| 46 |
verbose=True,
|
| 47 |
memory=ConversationBufferWindowMemory(k=2)
|