ferrarisf50 commited on
Commit
05d4f96
·
verified ·
1 Parent(s): be0cff7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -1
app.py CHANGED
@@ -60,7 +60,14 @@ temperature=0.5,
60
  model_id='HuggingFaceTB/SmolLM-135M-Instruct',# it is possible that this model may be overloaded
61
  custom_role_conversions=None,
62
  )'''
63
- model = TransformersModel(model_id="HuggingFaceTB/SmolLM-135M-Instruct")
 
 
 
 
 
 
 
64
 
65
  # Import tool from Hub
66
  image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
 
60
  model_id='HuggingFaceTB/SmolLM-135M-Instruct',# it is possible that this model may be overloaded
61
  custom_role_conversions=None,
62
  )'''
63
+ # model = TransformersModel(model_id="HuggingFaceTB/SmolLM-135M-Instruct")
64
+
65
+ model = HfApiModel(
66
+ max_tokens=2096,
67
+ temperature=0.5,
68
+ model_id='HuggingFaceTB/SmolLM-135M-Instruct',# it is possible that this model may be overloaded
69
+ custom_role_conversions=None,
70
+ )
71
 
72
  # Import tool from Hub
73
  image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)