Spaces:
Running
Running
| import google.generativeai as genai | |
| import gradio as gr | |
| import numpy as np | |
| import PIL.Image | |
| genai.configure(api_key="AIzaSyAj-b3sO_wUguMdpXWScxKzMHxb8C5cels") | |
| def ImageChat(image, prompt): | |
| # load model | |
| model = genai.GenerativeModel("gemini-pro-vision") | |
| # check image file and convert to a Numpy array | |
| if isinstance(image, np.ndarray): | |
| img = PIL.Image.fromarray(image) | |
| else: | |
| img = PIL.Image.open(image) | |
| response = model.generate_content([prompt, img]) | |
| return response.text | |
| app = gr.Interface(ImageChat, | |
| inputs = [gr.Image(label = "Image"), gr.Text(label = "Prompt")], | |
| outputs = gr.Text(label = "Response"), | |
| title = "Image Chat", | |
| theme = gr.themes.Soft()) | |
| app.launch() | |