viveknarayan commited on
Commit
8f0ddf2
1 Parent(s): e9de072

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -2,10 +2,9 @@ import numpy as np
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)
 
2
  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)