Spaces:
Runtime error
Runtime error
Update utility/utils.py
Browse files- utility/utils.py +1 -1
utility/utils.py
CHANGED
|
@@ -186,7 +186,7 @@ def Data_Extractor(data, client=client):
|
|
| 186 |
Output:
|
| 187 |
'''
|
| 188 |
# Call the API for inference
|
| 189 |
-
response = client.text_generation(text, max_new_tokens=1000
|
| 190 |
|
| 191 |
print("parse in text ---:",response)
|
| 192 |
|
|
|
|
| 186 |
Output:
|
| 187 |
'''
|
| 188 |
# Call the API for inference
|
| 189 |
+
response = client.text_generation(text, max_new_tokens=1000, temperature=0.4, top_k=50, top_p=0.9, repetition_penalty=1.2)
|
| 190 |
|
| 191 |
print("parse in text ---:",response)
|
| 192 |
|