Spaces:
Build error
Build error
update
Browse files
app.py
CHANGED
|
@@ -49,13 +49,13 @@ def text_generate(problem, template, prompt):
|
|
| 49 |
{
|
| 50 |
"top_p": 0.9,
|
| 51 |
"temperature": 1.1,
|
| 52 |
-
|
| 53 |
"return_full_text": False
|
| 54 |
}, "options":
|
| 55 |
{
|
| 56 |
"use_cache": True,
|
| 57 |
"wait_for_model":True
|
| 58 |
-
}}
|
| 59 |
response = requests.post(API_URL, headers=headers, json=json_)
|
| 60 |
print(f"Response is : {response}")
|
| 61 |
output = response.json()
|
|
|
|
| 49 |
{
|
| 50 |
"top_p": 0.9,
|
| 51 |
"temperature": 1.1,
|
| 52 |
+
"max_new_tokens": 250,
|
| 53 |
"return_full_text": False
|
| 54 |
}, "options":
|
| 55 |
{
|
| 56 |
"use_cache": True,
|
| 57 |
"wait_for_model":True
|
| 58 |
+
},}
|
| 59 |
response = requests.post(API_URL, headers=headers, json=json_)
|
| 60 |
print(f"Response is : {response}")
|
| 61 |
output = response.json()
|