Update app.py

app.py

@@ -2,14 +2,34 @@ import subprocess
 import sys
 import os
 
+# Fonction pour installer un package si non présent
+def install_package(package_name):
+    subprocess.run([sys.executable, "-m", "pip", "install", package_name], check=True)
+
 # Vérifiez si torch est installé, sinon installez-le
 try:
     import torch
 except ImportError:
     print("Torch n'est pas installé. Installation de torch...")
-    subprocess.run([sys.executable, "-m", "pip", "install", "torch"], check=True)
+    install_package("torch")
     import torch
 
+# Vérifiez si transformers est installé, sinon installez-le
+try:
+    from transformers import (
+        AutoModelForCausalLM,
+        AutoTokenizer,
+        TextIteratorStreamer,
+    )
+except ImportError:
+    print("Transformers n'est pas installé. Installation de transformers...")
+    install_package("transformers")
+    from transformers import (
+        AutoModelForCausalLM,
+        AutoTokenizer,
+        TextIteratorStreamer,
+    )
+
 # Installer flash-attn
 subprocess.run(
     "pip install flash-attn --no-build-isolation",
@@ -18,11 +38,6 @@ subprocess.run(
 )
 
 import gradio as gr
-from transformers import (
-    AutoModelForCausalLM,
-    AutoTokenizer,
-    TextIteratorStreamer,
-)
 from threading import Thread
 
 # Obtenir le token d'authentification Hugging Face