Spaces:
Build error
Build error
File size: 1,036 Bytes
541828e 3877213 f560b92 5ebef4a f560b92 3877213 5ebef4a 3877213 5ebef4a 3877213 5ebef4a |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 |
import os
import google.generativeai as genai
import gradio as gr
# Replace with your actual API key (ensure you are using the correct endpoint)
genai.configure(api_key=os.environ["GOOGLE_API_KEY"])
generation_config = {
"temperature": 0.7,
"top_p": 0.95,
"top_k": 64,
"max_output_tokens": 128,
"response_mime_type": "text/plain",
}
# Create the Gemini 1.5 Flash model
model = genai.GenerativeModel(
model_name="gemini-1.5-flash",
generation_config=generation_config,
)
# Define a function to generate responses
def generate_response(chatbot):
prompt = ""
for turn in chatbot:
prompt += f"{turn[0]}: {turn[1]}\n" # Format the conversation history
response = model.generate_text(
prompt=prompt,
temperature=0.7,
max_output_tokens=50,
)
return response.result
# Create Gradio interface
with gr.Blocks() as interface:
chatbot = gr.Chatbot()
msg = gr.Textbox()
clear = gr.ClearButton([msg, chatbot])
msg.submit(generate_response, [chatbot], chatbot)
interface.launch()
|