Update app.py

app.py

@@ -4,7 +4,7 @@ import gradio as gr
 import requests
 import pandas as pd
 import base64
-
+import ollama
 
 from typing import TypedDict, Annotated
 from langchain_core.messages import AnyMessage, SystemMessage, HumanMessage
@@ -13,13 +13,13 @@ from langchain.tools import Tool
 from langgraph.graph import START, StateGraph
 from langgraph.graph.message import add_messages
 from langgraph.prebuilt import ToolNode, tools_condition
-from langchain_huggingface import ChatHuggingFace, HuggingFaceEndpoint
+from langchain_ollama.chat_models import ChatOllama
 
 
 print("Loading LLM...")
-llm = HuggingFaceEndpoint(repo_id="meta-llama/Llama-3.2-3B-Instruct")
-chat = ChatHuggingFace(llm=llm)
-print(f"Model {chat.llm} downloaded!")
+ollama.pull('gemma3:4b')
+chat = ChatOllama(model="gemma3:4b", verbose=True)
+print(f"Model {chat.model} downloaded!")
 
 # (Keep Constants as is)
 # --- Constants ---