jdesiree commited on
Commit
dbbb319
·
verified ·
1 Parent(s): 56eab31

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -23,6 +23,7 @@ llm = HuggingFaceEndpoint(
23
  top_p=0.8,
24
  top_k=20,
25
  max_new_tokens=1024,
 
26
  huggingfacehub_api_token=os.getenv("HUGGINGFACEHUB_API_TOKEN")
27
  )
28
  math_template = ChatPromptTemplate.from_messages([
 
23
  top_p=0.8,
24
  top_k=20,
25
  max_new_tokens=1024,
26
+ task="conversational",
27
  huggingfacehub_api_token=os.getenv("HUGGINGFACEHUB_API_TOKEN")
28
  )
29
  math_template = ChatPromptTemplate.from_messages([