-
Notifications
You must be signed in to change notification settings - Fork 0
/
vision.py
40 lines (30 loc) · 1.1 KB
/
vision.py
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
from dotenv import load_dotenv
load_dotenv() # loading all the environment variables
import streamlit as st
import os
import google.generativeai as genai
from PIL import Image
genai.configure(api_key=os.getenv("GOOGLE_API_KEY"))
# function to load Gemini Pro model and get responses
model = genai.GenerativeModel("gemini-pro-vision")
def get_gemini_responses(input, image):
if input != "":
response = model.generate_content([input, image])
else:
response = model.generate_content(image)
return response.text
# Initialize our streamlit app
st.set_page_config(page_title = "Gemini Image Demo")
st.header("Gemini Application")
input = st.text_input("Input Prompt: ", key='input')
upload_file = st.file_uploader("Choose an Image: ", type=["jpg", "jpeg", "png"])
image = ""
if upload_file is not None:
image = Image.open(upload_file)
st.image(image, caption="Uploaded Image.", use_column_width=True)
submit = st.button("Tell me about the image")
## if submit is clicked
if submit:
response = get_gemini_responses(input, image)
st.subheader("The Response is")
st.write(response)