Spaces:
Sleeping
Sleeping
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")
|