Update app.py

app.py

@@ -3,7 +3,7 @@ import gradio as gr
 import requests
 import pandas as pd
 import base64
-
+import ollama
 from dotenv import load_dotenv
 from typing import TypedDict, Annotated
 from langchain_core.messages import AnyMessage, SystemMessage, HumanMessage
@@ -20,7 +20,10 @@ from huggingface_hub import login, InferenceClient
 # --- Constants ---
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
-model_id = "huggingface-projects/llama-2-13b-chat"
+print("Loading LLM...")
+ollama.pull('gemma3:4b')
+
+model_id = "google/gemma-3-4b-it"
 
 def get_file_path(task_id: str, question) -> str:
     """Retrieves reference file path."""