Update app.py

app.py

@@ -82,9 +82,14 @@ extract_text_tool = Tool(
 )
 
 # Generate the chat interface, including the tools
-print("Loading LLM...")
-ollama.pull('gemma3:4b')
-chat = ChatOllama(model="gemma3:4b", verbose=True)
+# print("Loading LLM...")
+# ollama.pull('gemma3:4b')
+from transformers import AutoProcessor, AutoModelForImageTextToText
+
+processor = AutoProcessor.from_pretrained("google/gemma-3-4b-it")
+model = AutoModelForImageTextToText.from_pretrained("google/gemma-3-4b-it")
+
+chat = ChatOllama(model=model, verbose=True)
 print(f"Model {chat.model} downloaded!")
 
 tools = [get_file_path_tool, get_content_tool, extract_text_tool, search_web_tool]