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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -10,8 +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
- 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 = """
 
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 = """