Spaces:
Sleeping
Sleeping
File size: 1,706 Bytes
86c6a74 |
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 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 |
# import required libraries
from dotenv import load_dotenv
load_dotenv() # to load all the env variables
import streamlit as st
import os
import google.generativeai as genai
from PIL import Image
import time
#-------------------------------------------#
genai.configure(api_key= os.getenv("GOOGLE_API_KEY"))
# creating function to load gemini pro model
model = genai.GenerativeModel("gemini-pro-vision")
model1 = genai.GenerativeModel("gemini-pro")
def get_response(input,image):
start_time = time.time()
if input != '' and image is not None:
response = model.generate_content([input,image])
elif image == None:
response = model1.generate_content(input)
else:
response = model.generate_content(image)
#response = response.parts
#for part in response:
#return (part.text)
end_time = time.time()
response_time = end_time - start_time
return response.text,response_time
# To set up streamlit
st.set_page_config(page_title="Content Generation LLM Model using Gemini")
st.header("Content Generation LLM Model ")
input = st.text_input("Input Prompt", key="input")
uploaded_file = st.file_uploader("Input Image", type=["jpg", "jpeg", "png"])
image=None
if uploaded_file is not None:
image = Image.open(uploaded_file)
st.image(image, caption="Uploaded Image.", use_column_width=True)
submit = st.button("Generate")
## when submit button is clicked,
if submit:
response_text,response_time = get_response(input,image)
st.subheader("The Generated Content:")
st.write(response_text)
st.write("Response Time :" ,response_time)
|