Mistral-7B-Summarizer-v2

Sleeping

Chan-Y commited on Jun 14

Commit

f6a07f3

•

1 Parent(s): a6ce7fc

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -15,7 +15,9 @@ llm = HuggingFaceEndpoint(
     repo_id="mistralai/Mistral-7B-Instruct-v0.3",
     task="text-generation",
     max_new_tokens=4096,
     do_sample=False,
 )
 llm_engine_hf = ChatHuggingFace(llm=llm)

     repo_id="mistralai/Mistral-7B-Instruct-v0.3",
     task="text-generation",
     max_new_tokens=4096,
+    temperature=0.5,
     do_sample=False,
+    task="summary_text"
 )
 llm_engine_hf = ChatHuggingFace(llm=llm)