pvanand commited on
Commit
2f3056d
·
verified ·
1 Parent(s): 7924068

Update routers/scraping_router.py

Browse files
Files changed (1) hide show
  1. routers/scraping_router.py +2 -1
routers/scraping_router.py CHANGED
@@ -110,7 +110,8 @@ async def generate_cheerio_script(
110
  response = ""
111
  response = ai_client.chat(
112
  prompt=user_prompt,
113
- system_message=system_prompt
 
114
  )
115
 
116
  cheerio_pattern = r'<cheerio_script>(.*?)</cheerio_script>'
 
110
  response = ""
111
  response = ai_client.chat(
112
  prompt=user_prompt,
113
+ system_message=system_prompt,
114
+ model_id="google/gemini-pro-1.5" #"deepseek/deepseek-chat"
115
  )
116
 
117
  cheerio_pattern = r'<cheerio_script>(.*?)</cheerio_script>'