|
|
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): |
|
|
|
|
|
|
|
|
model = genai.GenerativeModel("gemini-pro-vision") |
|
|
|
|
|
|
|
|
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(), gr.Text()], |
|
|
outputs = gr.Text(label = "Chat"), |
|
|
title = "Image Chat", |
|
|
theme = gr.themes.Soft()) |
|
|
|
|
|
app.launch() |
|
|
|
