Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -75,10 +75,10 @@ api_response_prompt = PromptTemplate(input_variables=['api_docs_mck',
|
|
75 |
# --model, memory object, and llm_chain
|
76 |
@cl.on_chat_start
|
77 |
def setup_multiple_chains():
|
78 |
-
llm = HuggingFaceHub(repo_id="google/gemma-2-2b-it"
|
79 |
-
temperature=0.7,
|
80 |
-
huggingface_api_token=HUGGINGFACEHUB_API_TOKEN,
|
81 |
-
device=device)
|
82 |
|
83 |
conversation_memory = ConversationBufferMemory(memory_key="chat_history",
|
84 |
max_len=200,
|
|
|
75 |
# --model, memory object, and llm_chain
|
76 |
@cl.on_chat_start
|
77 |
def setup_multiple_chains():
|
78 |
+
llm = HuggingFaceHub(repo_id="google/gemma-2-2b-it") #,
|
79 |
+
#temperature=0.7,
|
80 |
+
#huggingface_api_token=HUGGINGFACEHUB_API_TOKEN,
|
81 |
+
#device=device)
|
82 |
|
83 |
conversation_memory = ConversationBufferMemory(memory_key="chat_history",
|
84 |
max_len=200,
|