Horus7 commited on
Commit
4ddb435
1 Parent(s): 8eddb55

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +23 -1
app.py CHANGED
@@ -1,3 +1,25 @@
1
  import gradio as gr
 
2
 
3
- gr.load("models/Horus7/kaduce").launch()
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  import gradio as gr
2
+ import tensorflow as tf
3
 
4
+ model = tf.keras.models.load_model("models/Horus7/kaduce") # Charger le modèle TensorFlow
5
+
6
+ def predict(image):
7
+ # Prétraiter l'image (vous pouvez adapter cela en fonction des besoins de votre modèle)
8
+ preprocessed_image = preprocess_image(image)
9
+
10
+ # Faire une prédiction avec le modèle
11
+ prediction = model.predict(preprocessed_image)
12
+
13
+ # Renvoyer la prédiction
14
+ return prediction
15
+
16
+ # Fonction pour prétraiter l'image avant la prédiction
17
+ def preprocess_image(image):
18
+ # Effectuer les étapes de prétraitement nécessaires pour votre modèle (redimensionnement, normalisation, etc.)
19
+ return image
20
+
21
+ # Interface Gradio
22
+ inputs = gr.inputs.Image() # Entrée : une image
23
+ outputs = gr.outputs.Label() # Sortie : une étiquette
24
+
25
+ gr.Interface(fn=predict, inputs=inputs, outputs=outputs).launch()