imanolcb commited on
Commit
266edbe
·
verified ·
1 Parent(s): 7f68d92

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -1
app.py CHANGED
@@ -1,8 +1,13 @@
1
  import gradio as gr
2
  import tensorflow as tf
3
 
 
 
 
 
 
4
  # Cargar el modelo desde Hugging Face
5
- model = tf.keras.models.load_model("hf://imanolcb/basicFruitClassifier")
6
 
7
  # Preprocesamiento de las imágenes de entrada
8
  def preprocess_image(image):
 
1
  import gradio as gr
2
  import tensorflow as tf
3
 
4
+ import os
5
+ os.environ["KERAS_BACKEND"] = "jax"
6
+
7
+ import keras
8
+
9
  # Cargar el modelo desde Hugging Face
10
+ model = keras.saving.load_model("hf://imanolcb/basicFruitClassifier")
11
 
12
  # Preprocesamiento de las imágenes de entrada
13
  def preprocess_image(image):