GeminiPro / app.py
LOVEBHATTI's picture
Update app.py
105c824 verified
raw
history blame
No virus
841 Bytes
import streamlit as st
import google-generativeai as genAi
from PIL import Image
genAi.configure(api_key="AIzaSyBZQmot8_8bDxuGiffJ06woJCzH140Erc4str")
def get_response(inputt,image):
model=genAi.GenerativeModel('gemini-pro-vision')
if inputt!="":
response=model.generate_content([inputt,image])
else:
# response=model.generate_content([image])
response=model.generate_content(image)
return response.text
st.header("PROJECT")
input=st.text_input("Input Prompt",key='input')
upload_file=st.file_uploader("Choose yout file",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("ASK")
if submit:
response=get_response(input,image)
st.write(response)