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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
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)