SpaghettiM commited on
Commit
2a09e95
·
verified ·
1 Parent(s): 1b86922

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -3,7 +3,7 @@ import gradio as gr
3
  import requests
4
  import inspect
5
  import pandas as pd
6
- from smolagents import ToolCallingAgent, DuckDuckGoSearchTool, InferenceClientModel
7
 
8
  # (Keep Constants as is)
9
  # --- Constants ---
@@ -13,7 +13,7 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
13
  # ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
14
  class BasicAgent:
15
  def __init__(self):
16
- self.model = InferenceClientModel(model_id='google/gemma-7b'#'Qwen/Qwen3-8B-FP8'
17
  )
18
 
19
  self.agent = ToolCallingAgent(
 
3
  import requests
4
  import inspect
5
  import pandas as pd
6
+ from smolagents import ToolCallingAgent, DuckDuckGoSearchTool, InferenceClientModel, TransformersModel
7
 
8
  # (Keep Constants as is)
9
  # --- Constants ---
 
13
  # ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
14
  class BasicAgent:
15
  def __init__(self):
16
+ self.model = TransformersModel(model_id='google/gemma-7b'#'Qwen/Qwen3-8B-FP8'
17
  )
18
 
19
  self.agent = ToolCallingAgent(