camparchimedes commited on
Commit
5607ea3
ยท
verified ยท
1 Parent(s): 861654a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -89,7 +89,7 @@ api_response_prompt = PromptTemplate(
89
  def setup_multiple_chains():
90
 
91
  llm = HuggingFaceEndpoint(
92
- repo_id="RuterNorway/Llama-2-7b-chat-norwegian", #"norallm/normistral-7b-warm-instruct",
93
  huggingfacehub_api_token=HUGGINGFACEHUB_API_TOKEN,
94
  #max_new_tokens=512,
95
  temperature=0.7,
@@ -134,7 +134,7 @@ async def handle_message(message: cl.Message):
134
  #booked = is_booking_query(user_message)
135
  #if booked:
136
 
137
- if re.search(r'\b[A-Z]{6}\d{6}\b', user_message):
138
  response = await api_chain.acall(user_message,
139
  callbacks=[cl.AsyncLangchainCallbackHandler()])
140
 
 
89
  def setup_multiple_chains():
90
 
91
  llm = HuggingFaceEndpoint(
92
+ repo_id="google/gemma-2-2b-it", #"norallm/normistral-7b-warm-instruct",
93
  huggingfacehub_api_token=HUGGINGFACEHUB_API_TOKEN,
94
  #max_new_tokens=512,
95
  temperature=0.7,
 
134
  #booked = is_booking_query(user_message)
135
  #if booked:
136
 
137
+ if re.search(r'\b[A-Z]{6}\d{6}\b', user_message): # ex., "Hei, har du booking info for EQJLCQ362149?"
138
  response = await api_chain.acall(user_message,
139
  callbacks=[cl.AsyncLangchainCallbackHandler()])
140