Update app.py
Browse files
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
|
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."
|