Spaces:

Chan-Y
/

Labeler-Mistral

Sleeping

Chan-Y commited on Aug 1

Commit

1e5ee0a

•

1 Parent(s): 76f12fd

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -7,7 +7,7 @@ from langchain_huggingface import HuggingFaceEndpoint
 # Initialize the LLM and other components
 llm = HuggingFaceEndpoint(
     repo_id="mistralai/Mistral-7B-Instruct-v0.3",
-    task="text-generation",
     max_new_tokens=64,
     temperature=0.5,
     do_sample=False,

 # Initialize the LLM and other components
 llm = HuggingFaceEndpoint(
     repo_id="mistralai/Mistral-7B-Instruct-v0.3",
+    task="text-classification",
     max_new_tokens=64,
     temperature=0.5,
     do_sample=False,