Hugging Face's logo

Spaces:
ybelkada
/
i-like-flan-ul2
Running

App Files Community
7
ybelkada commited on
Commit
a5cf62b
•
1 Parent(s): 2263b4b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -9,7 +9,7 @@ UL2_API_URL = "https://api-inference.huggingface.co/models/google/flan-ul2"
9
  FLAN_API_URL = "https://api-inference.huggingface.co/models/google/flan-t5-xxl"
10
 
11
  headers = {"Authorization": f"Bearer {TOKEN}"}
12
- MAX_NEW_TOKENS = 128
13
 
14
  def query(text, api_url):
15
  response = requests.post(api_url, headers=headers, json={"inputs":text, "parameters": {"max_new_tokens":MAX_NEW_TOKENS}})
 
9
  FLAN_API_URL = "https://api-inference.huggingface.co/models/google/flan-t5-xxl"
10
 
11
  headers = {"Authorization": f"Bearer {TOKEN}"}
12
+ MAX_NEW_TOKENS = 256
13
 
14
  def query(text, api_url):
15
  response = requests.post(api_url, headers=headers, json={"inputs":text, "parameters": {"max_new_tokens":MAX_NEW_TOKENS}})