Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -1,3 +1,25 @@
|
|
1 |
import gradio as gr
|
|
|
2 |
|
3 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
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()
|