dongsiqie commited on
Commit
0ae56f8
·
1 Parent(s): d693eb1

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +25 -0
app.py ADDED
@@ -0,0 +1,25 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import gradio as gr
2
+ import os
3
+ import google.generativeai as genai
4
+
5
+ def generate_chat(api_key, prompt_parts):
6
+ genai.configure(api_key=api_key)
7
+ model = genai.GenerativeModel(model_name="gemini-pro")
8
+ response = model.generate_content([prompt_parts])
9
+ return response.text
10
+
11
+ with gr.Blocks() as app:
12
+ gr.Markdown("# gemini-pro generate_chat demo")
13
+ gr.Markdown("This is just a very simple demo,Enter your API key to start the conversation. Get a free API key from here: https://ai.google.dev/")
14
+ api_key_input = gr.Textbox(label="API Key", type="password", placeholder="Enter your API key", lines=1)
15
+ prompt_input = gr.Textbox(label="Prompt", placeholder="Enter your Prompt", lines=3)
16
+ generate_button = gr.Button("Generate Content")
17
+ output = gr.Markdown()
18
+
19
+ generate_button.click(
20
+ generate_chat,
21
+ inputs=[api_key_input, prompt_input],
22
+ outputs=output
23
+ )
24
+
25
+ app.launch()