NarmathaP commited on
Commit
48f7fd5
1 Parent(s): 43cd023

Upload 3 files

Browse files
Files changed (3) hide show
  1. .env +1 -0
  2. app.py +34 -0
  3. requirements.txt +3 -0
.env ADDED
@@ -0,0 +1 @@
 
 
1
+ GOOGLE_API_KEY="AIzaSyDsQ2F-6XfxuRFUXYFrcG6LSB95Z6xLre8"
app.py ADDED
@@ -0,0 +1,34 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from dotenv import load_dotenv
2
+ load_dotenv()
3
+
4
+ import streamlit as st
5
+ import os
6
+ import google.generativeai as genai
7
+ from PIL import Image
8
+
9
+ genai.configure(api_key=os.getenv("GOOGLE_API_KEY"))
10
+ model = genai.GenerativeModel("gemini-1.5-flash")
11
+ def get_gemini_response(input,image):
12
+ if input!="":
13
+ response = model.generate_content([input,image])
14
+ else:
15
+ response = model.generate_content(image)
16
+ return response.text
17
+
18
+ #Front end
19
+ st.set_page_config(page_title="Gemini Pro Image Recognition")
20
+ st.header("Gemini Pro Demo")
21
+ input = st.text_input("Input Prompt: ",key="input")
22
+
23
+ uploaded_file = st.file_uploader("Choose the image ",type=["jpg","jpeg","png"])
24
+ image = ""
25
+ if uploaded_file is not None:
26
+ image = Image.open(uploaded_file)
27
+ st.image(image,caption="Uploaded Image",use_column_width=True)
28
+
29
+ submit = st.button("Tell me about the picture that you see uploaded")
30
+
31
+ if submit:
32
+ response = get_gemini_response(input,image)
33
+ st.subheader("The response will be")
34
+ st.write(response)
requirements.txt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ streamlit
2
+ google-generativeai
3
+ python-dotenv