Update model.py

model.py

@@ -1,16 +1,26 @@
 import numpy as np
 import gradio as gr
 from PIL import Image
+import tensorflow as tf
 import keras
 from huggingface_hub import from_pretrained_keras
 
 
-model = from_pretrained_keras("keras-io/lowlight-enhance-mirnet", compile=False)
+model = from_pretrained_keras("hbpkillerX/image_enhancer", compile=False)
+
+def autocontrast(tensor, cutoff=0):
+    tensor = tf.cast(tensor, dtype=tf.float32)
+    min_val = tf.reduce_min(tensor)
+    max_val = tf.reduce_max(tensor)
+    range_val = max_val - min_val
+    adjusted_tensor = tf.clip_by_value(tf.cast(tf.round((tensor - min_val - cutoff) * (255 / (range_val - 2 * cutoff))), tf.uint8), 0, 255)
+    return adjusted_tensor
 
 
 
 def infer(original_image):
     image = keras.utils.img_to_array(original_image)
+    image = autocontrast(image)
     image = image.astype("float32") / 255.0
     image = np.expand_dims(image, axis=0)
     output = model.predict(image)