Update app.py

app.py

@@ -2,13 +2,15 @@ import gradio as gr
 from transformers import BlenderbotTokenizer, BlenderbotForConditionalGeneration
 import spaces 
 
-tokenizer = BlenderbotTokenizer.from_pretrained('meta-llama/Meta-Llama-3-8B-Instruct')
-model = BlenderbotForConditionalGeneration.from_pretrained("meta-llama/Meta-Llama-3-8B-Instruct")
+
+tokenizer = AutoTokenizer.from_pretrained("meta-llama/Meta-Llama-3-8B-Instruct")
+model = AutoModelForCausalLM.from_pretrained("meta-llama/Meta-Llama-3-8B-Instruct", device_map="auto")
 
 # Initialiser le générateur de texte avec GPT-2
 
 # Optionnel : Fixer une graine aléatoire pour la reproductibilit
-@spaces.GPU
+@spaces.GPU(duration=120)
+
 # Fonction de génération de texte
 def generate_text(prompt):