Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -9,10 +9,11 @@ st.set_page_config(page_title="Assistant Mathématique", page_icon="🔢", layou
|
|
9 |
def load_model():
|
10 |
"""Charge le modèle et le tokenizer (mise en cache par Streamlit)"""
|
11 |
model_name = "analist/deepseek-math-gguf" # Remplacez par votre nom de modèle
|
12 |
-
|
13 |
model = AutoModelForCausalLM.from_pretrained(
|
14 |
"analist/deepseek-math-gguf", model_file="model.gguf"
|
15 |
)
|
|
|
16 |
return model, tokenizer
|
17 |
|
18 |
def generate_response(prompt, model, tokenizer):
|
|
|
9 |
def load_model():
|
10 |
"""Charge le modèle et le tokenizer (mise en cache par Streamlit)"""
|
11 |
model_name = "analist/deepseek-math-gguf" # Remplacez par votre nom de modèle
|
12 |
+
|
13 |
model = AutoModelForCausalLM.from_pretrained(
|
14 |
"analist/deepseek-math-gguf", model_file="model.gguf"
|
15 |
)
|
16 |
+
tokenizer = AutoTokenizer.from_pretrained(model)
|
17 |
return model, tokenizer
|
18 |
|
19 |
def generate_response(prompt, model, tokenizer):
|