Update app.py

app.py

@@ -6,7 +6,6 @@ import matplotlib.pyplot as plt
 import numpy as np
 from huggingface_hub import login
 import os
-
 login(token=os.environ["HF_TOKEN"])
 
 # Liste des modèles
@@ -28,6 +27,7 @@ def load_model(model_name):
     model = AutoModelForCausalLM.from_pretrained(model_name, device_map="cpu")
     return f"Modèle {model_name} chargé avec succès sur CPU."
 
+@spaces.GPU(duration=300)
 def generate_text(input_text, temperature, top_p, top_k):
     global model, tokenizer