Update app.py

added gemini ai

app.py

@@ -6,6 +6,8 @@ import requests
 import torch
 import torchvision
 
+from langchain_google_genai import GoogleGenerativeAI
+
 from langchain.prompts import PromptTemplate
 from langchain.chains import LLMChain
 from langchain.chat_models import ChatOpenAI
@@ -17,8 +19,12 @@ from transformers import Pix2StructForConditionalGeneration, Pix2StructProcessor
 from transformers import BlipProcessor, BlipForConditionalGeneration
 import os
 
-os.environ["OPENAI_API_KEY"] = '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'
-llm = ChatOpenAI(temperature=0.2, model_name="gpt-3.5-turbo")
+# os.environ["OPENAI_API_KEY"] = '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'
+os.environ["GOOGLE_API_KEY"] = 'AIzaSyAsZTv6rUZq0TAh6yfmVCDA0tPIcGU3VxA'
+
+# llm = ChatOpenAI(temperature=0.2, model_name="gpt-3.5-turbo")
+llm = ChatGoogleGenerativeAI(google_api_key=google_api_key, model="gemini-pro")
+
 prompt = PromptTemplate(
     input_variables=["question", "elements"],
     template="""You are a helpful assistant that can answer question related to an image. You have the ability to see the image and answer questions about it.