Spaces:
Sleeping
Sleeping
fix model task
Browse files
app.py
CHANGED
@@ -37,6 +37,7 @@ translation_pipeline = pipeline(
|
|
37 |
def get_llm_hf_inference(model_id=model_id, max_new_tokens=128, temperature=temperature):
|
38 |
llm = HuggingFaceEndpoint(
|
39 |
repo_id=model_id,
|
|
|
40 |
max_new_tokens=max_new_tokens,
|
41 |
temperature=temperature,
|
42 |
token = os.getenv("HF_TOKEN")
|
|
|
37 |
def get_llm_hf_inference(model_id=model_id, max_new_tokens=128, temperature=temperature):
|
38 |
llm = HuggingFaceEndpoint(
|
39 |
repo_id=model_id,
|
40 |
+
task="text-generation",
|
41 |
max_new_tokens=max_new_tokens,
|
42 |
temperature=temperature,
|
43 |
token = os.getenv("HF_TOKEN")
|