viveknarayan commited on
Commit
12835b5
·
verified ·
1 Parent(s): 696013e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -3,8 +3,9 @@ import gradio as gr
3
  from PIL import Image
4
  import keras
5
  from huggingface_hub import from_pretrained_keras
 
6
 
7
- model = from_pretrained_keras("keras-io/lowlight-enhance-mirnet", compile=False)
8
 
9
  def infer(original_image):
10
  image = keras.utils.img_to_array(original_image)
 
3
  from PIL import Image
4
  import keras
5
  from huggingface_hub import from_pretrained_keras
6
+ import tf_keras as k3
7
 
8
+ model = k3.models.load_model('model.h5')
9
 
10
  def infer(original_image):
11
  image = keras.utils.img_to_array(original_image)