Update app.py

app.py

@@ -4,7 +4,7 @@ from transformers import pipeline
 from pptx import Presentation
 from pptx.util import Inches
 from transformers import AutoModelForCausalLM, AutoTokenizer, BitsAndBytesConfig, TextIteratorStreamer
-from llama_cpp import LLMCppModel
+from llama_cpp import Llama LLMCppModel
 
 
 # Préprompt pour donner des instructions au modèle
@@ -25,7 +25,9 @@ Voici le texte à transformer en présentation :"""
 #from llama_cpp import LLMCppModel
 model_path = "MisterAI/Bartowski_MistralAI_Mistral-Small-24B-Base-2501-GGUF"
 #text_to_presentation = LLMCppModel(model_path)
-text_to_presentation = model_path
+#text_to_presentation = model_path
+text_to_presentation = Llama(model_path, )
+text_to_presentation = Llama(model_path, verbose=True)
 
 # Chargement du tokenizer depuis Hugging Face
 #from transformers import AutoTokenizer