Hugging Face's logo

Spaces:
Fiqa
/
ChatWithData
Sleeping

App Files Community
Fiqa commited on
Commit
4aea096
·
verified ·
1 Parent(s): c82c449

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -34,7 +34,7 @@ login(token=HUGGINGFACE_API_KEY)
34
 
35
  hf_llm = HuggingFaceHub(
36
  repo_id="google/flan-t5-large",
37
- model_kwargs={"temperature": 0, "max_length": 64},
38
  huggingfacehub_api_token=HUGGINGFACE_API_KEY
39
  )
40
 
 
34
 
35
  hf_llm = HuggingFaceHub(
36
  repo_id="google/flan-t5-large",
37
+ model_kwargs={"temperature": 0, "max_length": 64, "task": "text2text-generation"},
38
  huggingfacehub_api_token=HUGGINGFACE_API_KEY
39
  )
40