aldohenrique commited on
Commit
931fbdb
·
verified ·
1 Parent(s): a968ffd

Update ai_logic.py

Browse files
Files changed (1) hide show
  1. ai_logic.py +7 -3
ai_logic.py CHANGED
@@ -18,12 +18,16 @@ HF_TOKEN = os.getenv("HF_TOKEN")
18
  if not HF_TOKEN:
19
  raise ValueError("Token HF_TOKEN não encontrado")
20
 
 
21
  MODELS = {
22
- "Phi-3 Mini": "microsoft/Phi-3-mini-4k-instruct",
23
  "Mistral 7B": "mistralai/Mistral-7B-Instruct-v0.3",
24
- "Deepseek (chat) 7B":"deepseek-ai/DeepSeek-V2-Lite-Chat",
25
- "Gemma (Google) 7B":"google/gemma-7b"
 
 
26
  }
 
 
27
  DEFAULT_MODEL = "Phi-3 Mini"
28
 
29
  # --- Gerenciamento de Sessão ---
 
18
  if not HF_TOKEN:
19
  raise ValueError("Token HF_TOKEN não encontrado")
20
 
21
+
22
  MODELS = {
 
23
  "Mistral 7B": "mistralai/Mistral-7B-Instruct-v0.3",
24
+ "Phi-3 Mini (Microsoft)": "microsoft/Phi-3-mini-4k-instruct",
25
+ "Deepseek (chat) 7B": "deepseek-ai/deepseek-vl-7b-chat",
26
+ "Gemma 7B (Google)":"google/gemma-7b-it",
27
+ "Zephyr 7B": "HuggingFaceH4/zephyr-7b-beta"
28
  }
29
+
30
+
31
  DEFAULT_MODEL = "Phi-3 Mini"
32
 
33
  # --- Gerenciamento de Sessão ---