Google Gemini API Python Example

import vertexai
from vertexai.preview.generative_models import GenerativeModel, Part

model = GenerativeModel("gemini-pro")
responses = model.generate_content(
"""Write a meal plan for today""",
generation_config={
"max_output_tokens": 2048,
"temperature": 0.9,
"top_p": 1
},
stream=True,
)

for response in responses:
print(response.candidates[0].content.parts[0].text)

Gradio UI

import gradio as gr
import vertexai
from vertexai.preview.generative_models import GenerativeModel, Part

def generate(prompt):
model = GenerativeModel("gemini-pro")
response = model.generate_content(
prompt,
generation_config={
"max_output_tokens": 2048,
"temperature": 0.9,
"top_p": 1
},
stream=False,
)
output = response.candidates[0].content.parts[0].text
return output

iface = gr.Interface(
fn=generate,
inputs="text",
outputs="markdown",
title="Gemini Pro API UI"
)

iface.launch()