Spaces:
Running
Running
Update ai_logic.py
Browse files- ai_logic.py +5 -4
ai_logic.py
CHANGED
|
@@ -20,13 +20,14 @@ if not HF_TOKEN:
|
|
| 20 |
|
| 21 |
|
| 22 |
MODELS = {
|
| 23 |
-
"Mistral 7B": "mistralai/Mistral-7B-Instruct-v0.3",
|
| 24 |
-
"Phi-3 Mini (
|
| 25 |
-
"Zephyr 7B (Teste)": "HuggingFaceH4/zephyr-7b-beta"
|
|
|
|
| 26 |
}
|
| 27 |
|
| 28 |
|
| 29 |
-
DEFAULT_MODEL = "
|
| 30 |
|
| 31 |
# --- Gerenciamento de Sessão ---
|
| 32 |
user_sessions: Dict[str, Dict[str, List | Dict]] = {} # {session_id: {'conversation': [], 'user_profile': {}}}
|
|
|
|
| 20 |
|
| 21 |
|
| 22 |
MODELS = {
|
| 23 |
+
"Mistral 7B (Mais acertivo)": "mistralai/Mistral-7B-Instruct-v0.3",
|
| 24 |
+
"Phi-3 Mini (Mais rápido)": "microsoft/Phi-3-mini-4k-instruct",
|
| 25 |
+
"Zephyr 7B (Teste)": "HuggingFaceH4/zephyr-7b-beta",
|
| 26 |
+
"Phi-3 Medium (Meio Termo)":"microsoft/Phi-3-medium-4k-instruct"
|
| 27 |
}
|
| 28 |
|
| 29 |
|
| 30 |
+
DEFAULT_MODEL = "Mistral 7B (Mais acertivo)"
|
| 31 |
|
| 32 |
# --- Gerenciamento de Sessão ---
|
| 33 |
user_sessions: Dict[str, Dict[str, List | Dict]] = {} # {session_id: {'conversation': [], 'user_profile': {}}}
|