E-Adam commited on
Commit
17dc88e
·
verified ·
1 Parent(s): b4b0009

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +30 -1
app.py CHANGED
@@ -5,8 +5,37 @@ os.environ['CUDA_VISIBLE_DEVICES'] = '-1'
5
  #subprocess.check_call([sys.executable, "-m", "pip", "install", "tensorflow-cpu"])
6
 
7
  import gradio as gr
8
-
9
  import numpy as np
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
10
  import tensorflow as tf
11
  #from tensorflow.keras.models import Sequential
12
  #from tensorflow.keras.layers import Embedding, GRU, Dense
 
5
  #subprocess.check_call([sys.executable, "-m", "pip", "install", "tensorflow-cpu"])
6
 
7
  import gradio as gr
 
8
  import numpy as np
9
+
10
+ def setup_tensorflow():
11
+ try:
12
+ import tensorflow as tf
13
+ print(f"TensorFlow {tf.__version__} déjà disponible")
14
+ return tf
15
+ except ImportError:
16
+ print("Installation de TensorFlow...")
17
+ try:
18
+ # Installation spécifique pour environnement Gradio
19
+ subprocess.check_call([
20
+ sys.executable, "-m", "pip", "install",
21
+ "tensorflow-cpu==2.13.0",
22
+ "--no-deps", "--force-reinstall", "--quiet"
23
+ ])
24
+
25
+ # Clear import cache
26
+ if 'tensorflow' in sys.modules:
27
+ del sys.modules['tensorflow']
28
+
29
+ import tensorflow as tf
30
+ print(f"TensorFlow {tf.__version__} installé avec succès")
31
+ return tf
32
+
33
+ except Exception as e:
34
+ print(f"Erreur d'installation: {e}")
35
+ return None
36
+
37
+ # Usage dans votre app Gradio
38
+ tf = setup_tensorflow()
39
  import tensorflow as tf
40
  #from tensorflow.keras.models import Sequential
41
  #from tensorflow.keras.layers import Embedding, GRU, Dense