analist commited on
Commit
4085a4c
·
verified ·
1 Parent(s): 6517c51

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
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
- tokenizer = AutoTokenizer.from_pretrained(model_name)
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):