Spaces:
Sleeping
Sleeping
Update palmapi.py
Browse files- palmapi.py +2 -4
palmapi.py
CHANGED
|
@@ -35,16 +35,14 @@ def gen_text(
|
|
| 35 |
parameters=None
|
| 36 |
):
|
| 37 |
if parameters is None:
|
| 38 |
-
model = 'models/text-bison-001'
|
| 39 |
temperature = 0.7
|
| 40 |
-
candidate_count = 1
|
| 41 |
top_k = 40
|
| 42 |
top_p = 0.95
|
| 43 |
|
| 44 |
parameters = {
|
| 45 |
-
'model':
|
|
|
|
| 46 |
'temperature': temperature,
|
| 47 |
-
'candidate_count': candidate_count,
|
| 48 |
'top_k': top_k,
|
| 49 |
'top_p': top_p,
|
| 50 |
}
|
|
|
|
| 35 |
parameters=None
|
| 36 |
):
|
| 37 |
if parameters is None:
|
|
|
|
| 38 |
temperature = 0.7
|
|
|
|
| 39 |
top_k = 40
|
| 40 |
top_p = 0.95
|
| 41 |
|
| 42 |
parameters = {
|
| 43 |
+
'model': 'models/text-bison-001',
|
| 44 |
+
'candidate_count': 1,
|
| 45 |
'temperature': temperature,
|
|
|
|
| 46 |
'top_k': top_k,
|
| 47 |
'top_p': top_p,
|
| 48 |
}
|