Pijush2023 commited on
Commit
8d76824
·
verified ·
1 Parent(s): 3075e32

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -63,7 +63,7 @@ entity_prompt = ChatPromptTemplate.from_messages([
63
  ])
64
 
65
  #chat_model = ChatOpenAI(temperature=0, model_name="gpt-4o", api_key=os.environ['OPENAI_API_KEY'])
66
- entity_chain = entity_prompt | chat_model.with_structured_output(Entities)
67
 
68
  def remove_lucene_chars(input: str) -> str:
69
  return input.translate(str.maketrans({
 
63
  ])
64
 
65
  #chat_model = ChatOpenAI(temperature=0, model_name="gpt-4o", api_key=os.environ['OPENAI_API_KEY'])
66
+ entity_chain = entity_prompt | llm.with_structured_output(Entities)
67
 
68
  def remove_lucene_chars(input: str) -> str:
69
  return input.translate(str.maketrans({