Update app.py
Browse files
app.py
CHANGED
@@ -40,12 +40,12 @@ from vertexai.preview.generative_models import GenerativeModel
|
|
40 |
generation_model = GenerativeModel(config.model)
|
41 |
|
42 |
def invoke(prompt):
|
43 |
-
completion = generation_model.predict(prompt = prompt,
|
44 |
-
|
45 |
-
|
46 |
-
|
47 |
-
|
48 |
-
|
49 |
completion = generation_model.generate_content(prompt, generation_config = {
|
50 |
"max_output_tokens": config.max_output_tokens,
|
51 |
"temperature": config.temperature,
|
|
|
40 |
generation_model = GenerativeModel(config.model)
|
41 |
|
42 |
def invoke(prompt):
|
43 |
+
#completion = generation_model.predict(prompt = prompt,
|
44 |
+
# max_output_tokens = config.max_output_tokens,
|
45 |
+
# temperature = config.temperature,
|
46 |
+
# top_k = config.top_k,
|
47 |
+
# top_p = config.top_p,
|
48 |
+
# ).text
|
49 |
completion = generation_model.generate_content(prompt, generation_config = {
|
50 |
"max_output_tokens": config.max_output_tokens,
|
51 |
"temperature": config.temperature,
|