Pijush2023 commited on
Commit
41d5ef2
·
verified ·
1 Parent(s): 5c32ce1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -34,7 +34,7 @@ entity_prompt = ChatPromptTemplate.from_messages([
34
  ("human", "Use the given format to extract information from the following input: {question}"),
35
  ])
36
 
37
- chat_model = ChatOpenAI(temperature=0, model_name="gpt-4o", api_key=os.environ['OPENAI_API_KEY']")
38
  entity_chain = entity_prompt | chat_model.with_structured_output(Entities)
39
 
40
  def remove_lucene_chars(input: str) -> str:
 
34
  ("human", "Use the given format to extract information from the following input: {question}"),
35
  ])
36
 
37
+ chat_model = ChatOpenAI(temperature=0, model_name="gpt-4o", api_key=os.environ['OPENAI_API_KEY'])
38
  entity_chain = entity_prompt | chat_model.with_structured_output(Entities)
39
 
40
  def remove_lucene_chars(input: str) -> str: