bstraehle commited on
Commit
22f9400
·
1 Parent(s): 0681a72

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -6
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
- 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,
 
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,