viveknarayan commited on
Commit
74ba37e
·
verified ·
1 Parent(s): 263095c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -2,9 +2,10 @@ import numpy as np
2
  import gradio as gr
3
  from PIL import Image
4
  import keras
 
5
  from huggingface_hub import from_pretrained_keras
6
 
7
- model = keras.models.load_model('model.h5')
8
 
9
  def infer(original_image):
10
  image = keras.utils.img_to_array(original_image)
 
2
  import gradio as gr
3
  from PIL import Image
4
  import keras
5
+ import tensorflow as tf
6
  from huggingface_hub import from_pretrained_keras
7
 
8
+ model = tf.keras.models.load_model('model.h5')
9
 
10
  def infer(original_image):
11
  image = keras.utils.img_to_array(original_image)