imravi commited on
Commit
2259de4
·
verified ·
1 Parent(s): afa36df

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -154,7 +154,7 @@ chatbot = PlantChatbot("/home/avinashhn/bheri_bot/ravi/new/plant_data_chunks_and
154
  async def response_generator(query: str):
155
  """Wrapper generator for streaming response"""
156
  async for chunk in chatbot.generate_response(query):
157
- yield f"data: {chunk}\n\n"
158
 
159
  @app.post("/chat")
160
  async def chat(query: Query):
 
154
  async def response_generator(query: str):
155
  """Wrapper generator for streaming response"""
156
  async for chunk in chatbot.generate_response(query):
157
+ yield f"chunk"
158
 
159
  @app.post("/chat")
160
  async def chat(query: Query):