lesimoes commited on
Commit
e9a1964
·
1 Parent(s): bf3be3c

fix model task

Browse files
Files changed (1) hide show
  1. app.py +1 -0
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")