Raffaele Terribile commited on
Commit
2bd2f31
·
unverified ·
1 Parent(s): 3fbdce0

Cambia modello di inferenza

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -27,8 +27,8 @@ def invert_sentence(sentence: str) -> str:
27
  class FirstAgent:
28
  ### First Agent is the first attempt to develop an agent for the course. ###
29
  def __init__(self):
30
- model_id = "HuggingFaceTB/SmolLM-135M-Instruct"
31
- model = TransformersModel(model_id=model_id, max_new_tokens=2048)
32
 
33
  # Inizializza l'agente
34
  self.agent = CodeAgent(
 
27
  class FirstAgent:
28
  ### First Agent is the first attempt to develop an agent for the course. ###
29
  def __init__(self):
30
+ model_id = "HuggingFaceTB/SmolLM2-1.7B-Instruct"
31
+ model = TransformersModel(model_id=model_id)
32
 
33
  # Inizializza l'agente
34
  self.agent = CodeAgent(