Spaces:
Running
Running
Update ai_logic.py
Browse files- 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 |
-
"
|
25 |
-
"
|
|
|
|
|
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 ---
|