pvanand commited on
Commit
1e480e3
1 Parent(s): 2bb5179

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -1
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="meta-llama/llama-3-70b-instruct"):
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")