Update app.py

app.py

@@ -1,5 +1,5 @@
 import streamlit as st
-from transformers import AutoModelForCausalLM, AutoTokenizer
+from ctransformers import AutoModelForCausalLM, AutoTokenizer
 import torch
 
 # Configuration de la page Streamlit
@@ -11,10 +11,8 @@ def load_model():
     model_name = "analist/deepseek-math-gguf"  # Remplacez par votre nom de modèle
     tokenizer = AutoTokenizer.from_pretrained(model_name)
     model = AutoModelForCausalLM.from_pretrained(
-        model_name,
-        torch_dtype=torch.float32,
-        device_map="cpu"
-    )
+    "analist/deepseek-math-gguf", model_file="model.gguf"
+)
     return model, tokenizer
 
 def generate_response(prompt, model, tokenizer):