shikharyashmaurya commited on
Commit
af44d27
·
verified ·
1 Parent(s): bcf0c8a

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +26 -0
app.py ADDED
@@ -0,0 +1,26 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import streamlit as st
2
+ import os
3
+ from PIL import Image
4
+ import google.generativeai as genai
5
+ secret_key = os.getenv("SECRET_KEY")
6
+ genai.configure(api_key=secret_key)
7
+ def get_gemini_response(input,image):
8
+ model = genai.GenerativeModel('gemini-pro-vision')
9
+ if input!="":
10
+ response = model.generate_content([input,image])
11
+ else:
12
+ response = model.generate_content(image)
13
+ return response.text
14
+ st.set_page_config(page_title="Image Summary")
15
+ st.header("Application")
16
+ input=st.text_input("Input Prompt: ",key="input")
17
+ uploaded_file = st.file_uploader("Choose an image...", type=["jpg", "jpeg", "png"])
18
+ image=""
19
+ if uploaded_file is not None:
20
+ image = Image.open(uploaded_file)
21
+ st.image(image, caption="Uploaded Image.", use_column_width=True)
22
+ submit=st.button("Tell me about the image")
23
+ if submit:
24
+ response=get_gemini_response(input,image)
25
+ st.subheader("The Response is")
26
+ st.write(response)