Spaces:

dpechenev
/

low_light_image_enhancement

Configuration error

Danila-Pechenev commited on Dec 13, 2022

Commit

7094aa3

1 Parent(s): 5182a5c

Change `create_model`

Files changed (1) hide show

app/model.py CHANGED Viewed

@@ -1,3 +1,4 @@
 from tensorflow import keras
 from PIL import Image
 import numpy as np
@@ -5,7 +6,7 @@ import io
 def create_model() -> keras.Model:
-    return keras.models.load_model("model")
 def run_model(image: io.BytesIO, model: keras.Model) -> Image.Image:

+from huggingface_hub import from_pretrained_keras
 from tensorflow import keras
 from PIL import Image
 import numpy as np
 def create_model() -> keras.Model:
+    return from_pretrained_keras("keras-io/lowlight-enhance-mirnet")
 def run_model(image: io.BytesIO, model: keras.Model) -> Image.Image: