jdesiree commited on
Commit
62075fe
·
verified ·
1 Parent(s): 5c95c43

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -22,7 +22,7 @@ llm = HuggingFaceEndpoint(
22
  temperature=0.7,
23
  max_new_tokens=512,
24
  huggingfacehub_api_token=os.getenv("HUGGINGFACEHUB_API_TOKEN"),
25
- task="text-generation" # <— important: use text-generation, not text2text-generation
26
  )
27
 
28
  math_template = ChatPromptTemplate.from_messages([
 
22
  temperature=0.7,
23
  max_new_tokens=512,
24
  huggingfacehub_api_token=os.getenv("HUGGINGFACEHUB_API_TOKEN"),
25
+ task="conversational"
26
  )
27
 
28
  math_template = ChatPromptTemplate.from_messages([