bstraehle commited on
Commit
2008c2a
·
1 Parent(s): 6919ace

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -37,7 +37,7 @@ generation_model = GenerativeModel("gemini-pro")
37
 
38
  def invoke(prompt):
39
  #completion = generation_model.predict(prompt = prompt).text
40
- completion = generation_model.generate_content(prompt = prompt).text
41
  wandb.log({"prompt": prompt, "completion": completion})
42
  return completion
43
  #return "Execution is commented out, but you can view the code at https://huggingface.co/spaces/bstraehle/google-vertex-ai-llm/tree/main."
 
37
 
38
  def invoke(prompt):
39
  #completion = generation_model.predict(prompt = prompt).text
40
+ completion = generation_model.generate_content(prompt).text
41
  wandb.log({"prompt": prompt, "completion": completion})
42
  return completion
43
  #return "Execution is commented out, but you can view the code at https://huggingface.co/spaces/bstraehle/google-vertex-ai-llm/tree/main."