[ADD] Choice of model type from UI

agent.py

@@ -41,9 +41,7 @@ class FinalAgent:
             chat = ChatHuggingFace(llm=llm, verbose=True)
         elif model_type == "OLLAMA":
             from langchain_ollama import ChatOllama
-            #chat = ChatOllama(model = "qwen2.5:14b-instruct")
             chat = ChatOllama(model = "qwen3:8b")
-            #chat = ChatOllama(model = "gpt-oss:20b")
         elif model_type == "GOOGLE":
             from langchain_google_genai import ChatGoogleGenerativeAI
             from langchain_core.rate_limiters import InMemoryRateLimiter
@@ -55,7 +53,8 @@ class FinalAgent:
                         max_bucket_size=10,  # Controls the maximum burst size.
                     )
             chat = ChatGoogleGenerativeAI(model="gemini-2.5-flash", rate_limiter=rate_limiter)
-
+        else:
+            raise ValueError(f'Model provider can be only one between GOOGLE, OLLAMA or HUGGINGFACE, received {model_type}')
 
         tools = [webpage_reader_tool,
                 transcribe_youtube_video_tool,

app.py

@@ -10,7 +10,7 @@ from agent import FinalAgent
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
 
-def run_and_submit_all( profile: gr.OAuthProfile | None):
+def run_and_submit_all( profile: gr.OAuthProfile | None, model_type):
     """
     Fetches all questions, runs the BasicAgent on them, submits all answers,
     and displays the results.
@@ -31,7 +31,7 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
 
     # 1. Instantiate Agent ( modify this part to create your agent)
     try:
-        agent = FinalAgent(model_type="GOOGLE")
+        agent = FinalAgent(model_type=model_type.upper().replace(' ', ''))
     except Exception as e:
         print(f"Error instantiating agent: {e}")
         return f"Error initializing agent: {e}", None
@@ -166,7 +166,7 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
 
 # --- Build Gradio Interface using Blocks ---
 with gr.Blocks() as demo:
-    gr.Markdown("# Basic Agent Evaluation Runner")
+    gr.Markdown("# Agent Evaluation Runner")
     gr.Markdown(
         """
         **Instructions:**
@@ -177,10 +177,12 @@ with gr.Blocks() as demo:
 
         ---
         **Disclaimers:**
-        Once clicking on the "submit button, it can take quite some time ( this is the time for the agent to go through all the questions).
-        This space provides a basic setup and is intentionally sub-optimal to encourage you to develop your own, more robust solution. For instance for the delay process of the submit button, a solution could be to cache the answers and submit in a seperate action or even to answer the questions in async.
+        Once clicking on the "submit" button, it can take quite some time (this is the time for the agent to go through all the questions).
         """
     )
+    model_selector = gr.Dropdown(
+            choices=["Google", "Ollama", "Hugging Face"], label="LLM provider", info="Which model provider should be used for inference?", interactive=True
+        )
 
     gr.LoginButton()
 
@@ -192,6 +194,7 @@ with gr.Blocks() as demo:
 
     run_button.click(
         fn=run_and_submit_all,
+        inputs=model_selector,
         outputs=[status_output, results_table]
     )