Update app.py

app.py

@@ -1,5 +1,13 @@
-# Load model directly
-from transformers import AutoTokenizer, AutoModelForCausalLM
+from transformers import pipeline
 
-tokenizer = AutoTokenizer.from_pretrained("meta-llama/Meta-Llama-3-8B")
-model = AutoModelForCausalLM.from_pretrained("meta-llama/Meta-Llama-3-8B")
+# Charger le pipeline pour la génération de texte
+pipe = pipeline("text-generation", model="meta-llama/Meta-Llama-3-8B")
+
+# Texte d'entrée pour démarrer la génération
+texte_entree = "Dans une forêt sombre, un"
+
+# Générer la suite du texte
+suite_texte = pipe(texte_entree, max_length=50, do_sample=True, temperature=0.7)
+
+# Afficher la suite du texte généré
+print(suite_texte[0]['generated_text'])