Mistral-7B-Summarizer-v2

Sleeping

Chan-Y commited on Jun 14

Commit

7f399fc

•

1 Parent(s): f6a07f3

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -13,11 +13,10 @@ from langchain_huggingface import ChatHuggingFace, HuggingFaceEndpoint
 llm = HuggingFaceEndpoint(
     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)

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