updated to llama3.2

This commit is contained in:
ShubhamSaboo 2024-10-11 21:48:03 -05:00
parent 2a030a1123
commit d077bb39a4

View file

@ -4,12 +4,12 @@ from scrapegraphai.graphs import SmartScraperGraph
# Set up the Streamlit app # Set up the Streamlit app
st.title("Web Scrapping AI Agent 🕵️‍♂️") st.title("Web Scrapping AI Agent 🕵️‍♂️")
st.caption("This app allows you to scrape a website using OpenAI API") st.caption("This app allows you to scrape a website using Llama 3.2")
# Set up the configuration for the SmartScraperGraph # Set up the configuration for the SmartScraperGraph
graph_config = { graph_config = {
"llm": { "llm": {
"model": "ollama/llama3", "model": "ollama/llama3.2",
"temperature": 0, "temperature": 0,
"format": "json", # Ollama needs the format to be specified explicitly "format": "json", # Ollama needs the format to be specified explicitly
"base_url": "http://localhost:11434", # set Ollama URL "base_url": "http://localhost:11434", # set Ollama URL