Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -1,8 +1,10 @@
|
|
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)
|
|
|
1 |
import gradio as gr
|
|
|
2 |
|
|
|
3 |
|
4 |
+
model_name = "Horus7/kaduce" # Remplacez par le nom de votre modèle Hugging Face
|
5 |
+
|
6 |
+
tokenizer = AutoTokenizer.from_pretrained(model_name)
|
7 |
+
model = AutoModelForSequenceClassification.from_pretrained(model_name)
|
8 |
def predict(image):
|
9 |
# Prétraiter l'image (vous pouvez adapter cela en fonction des besoins de votre modèle)
|
10 |
preprocessed_image = preprocess_image(image)
|