Danielrahmai1991 commited on
Commit
4396e10
1 Parent(s): 5d43e70

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -106,7 +106,7 @@ class Item(BaseModel):
106
 
107
  @app.post("/prompt/")
108
  def create_item(item: Item):
109
- message_response = llm_chain_model.run(item.get('question'))
110
  return {"item": item, "message": "LLm response", 'response': message_response}
111
 
112
 
 
106
 
107
  @app.post("/prompt/")
108
  def create_item(item: Item):
109
+ message_response = llm_chain_model.run(item.question)
110
  return {"item": item, "message": "LLm response", 'response': message_response}
111
 
112