Marylene commited on
Commit
fb0c9fa
·
verified ·
1 Parent(s): 75fc807

ajout fallback models oubliés

Browse files
Files changed (1) hide show
  1. quick_deploy_agent.py +8 -0
quick_deploy_agent.py CHANGED
@@ -5,6 +5,14 @@ import requests
5
  from smolagents import Tool, CodeAgent, InferenceClientModel
6
  from sentence_transformers import SentenceTransformer, util
7
 
 
 
 
 
 
 
 
 
8
  # ---- Mini référentiel COICOP (démo) ----
9
  COICOP_ITEMS = [
10
  {"code": "01.1.4.5.1", "label": "Laits caillés, fromage blanc, petites crèmes fromagères"},
 
5
  from smolagents import Tool, CodeAgent, InferenceClientModel
6
  from sentence_transformers import SentenceTransformer, util
7
 
8
+ FALLBACK_MODELS = [
9
+ # ordre de préférence ; tous dispos en Inference API publique
10
+ "meta-llama/Meta-Llama-3.1-8B-Instruct",
11
+ "mistralai/Mistral-7B-Instruct-v0.3",
12
+ "Qwen/Qwen2.5-7B-Instruct",
13
+ ]
14
+
15
+
16
  # ---- Mini référentiel COICOP (démo) ----
17
  COICOP_ITEMS = [
18
  {"code": "01.1.4.5.1", "label": "Laits caillés, fromage blanc, petites crèmes fromagères"},