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 |
|