Spaces:
Sleeping
Sleeping
Merge branch 'main' of https://huggingface.co/spaces/bziiit/VEGETALIS_AI_API
Browse files
main.py
CHANGED
@@ -273,7 +273,7 @@ def generate_answer(user_input: UserInput):
|
|
273 |
index=index)
|
274 |
|
275 |
if user_input.stream:
|
276 |
-
return StreamingResponse(stream_generator(answer,prompt_formated,
|
277 |
|
278 |
return {
|
279 |
"prompt": prompt_formated,
|
|
|
273 |
index=index)
|
274 |
|
275 |
if user_input.stream:
|
276 |
+
return StreamingResponse(stream_generator(answer,prompt_formated,infos_added_to_kb), media_type="application/json")
|
277 |
|
278 |
return {
|
279 |
"prompt": prompt_formated,
|