-
Notifications
You must be signed in to change notification settings - Fork 0
/
vision.py
40 lines (29 loc) · 1.06 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
#loading the environment variables
load_dotenv()
import streamlit as st
import os
import google.generativeai as genai
from PIL import Image
genai.configure(api_key=os.getenv("GOOGLE_API_KEY"))
model=genai.GenerativeModel("gemini-pro-vision")
def get_gemini_response(input,image):
if input!="":
response=model.generate_content([input,image])
else:
response=model.generate_content(image)
return response.text
#Streamlit page
st.set_page_config(page_title="Gemini Pro LLM Image Model")
st.header("Gemini Pro Vision Application")
input=st.text_input("Input: ", key="input")
uploaded_file=st.file_uploader("Choose an image from the device.",type=["jpg","jpeg","png"])
image=""
if uploaded_file is not None:
image=Image.open(uploaded_file)
st.image(image, caption="Uploaded Image",use_column_width=True)
submit=st.button("Tell me something about the image")
if submit:
response=get_gemini_response(input,image)
st.subheader("The response is")
st.write(response)