Spaces:
Running
Running
Update main.py
Browse files
main.py
CHANGED
@@ -252,7 +252,7 @@ async def news_assistant(query: NewsQueryModel, api_key: str = Depends(verify_ap
|
|
252 |
raise HTTPException(status_code=500, detail="Failed to fetch news data")
|
253 |
|
254 |
def process_response():
|
255 |
-
for content in chat_with_llama_stream(messages, model="
|
256 |
yield content
|
257 |
|
258 |
return StreamingResponse(process_response(), media_type="text/event-stream")
|
|
|
252 |
raise HTTPException(status_code=500, detail="Failed to fetch news data")
|
253 |
|
254 |
def process_response():
|
255 |
+
for content in chat_with_llama_stream(messages, model="google/gemini-pro-1.5"):
|
256 |
yield content
|
257 |
|
258 |
return StreamingResponse(process_response(), media_type="text/event-stream")
|