yashasgupta commited on
Commit
7c26f4d
1 Parent(s): d70c44b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -10,7 +10,8 @@ from langchain.prompts import ChatPromptTemplate
10
  os.environ["OPENAI_API_KEY"] = os.getenv("k3")
11
 
12
  # Initialize OpenAI with LangChain
13
- llm = OpenAI(api_key=os.environ["OPENAI_API_KEY"], model="cyberagent/llava-calm2-siglip", temperature=0)
 
14
 
15
  # Prompt template for image captioning
16
  caption_template = """
 
10
  os.environ["OPENAI_API_KEY"] = os.getenv("k3")
11
 
12
  # Initialize OpenAI with LangChain
13
+ #llm = OpenAI(api_key=os.environ["OPENAI_API_KEY"], model="cyberagent/llava-calm2-siglip", temperature=0)
14
+ llm = OpenAI(api_key=os.environ["OPENAI_API_KEY"], model="gpt-4-vision-preview", temperature=0)
15
 
16
  # Prompt template for image captioning
17
  caption_template = """