Spaces:
Running
Running
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):
|