camparchimedes commited on
Commit
29dfc22
ยท
verified ยท
1 Parent(s): a9b6b8d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
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,