initial commit

app.py

@@ -36,7 +36,7 @@ def convert_pil_to_cv2(image):
     return open_cv_image
 
 
-def main(image):
+def upscale(image):
     model_path = "models/model.ort"
     img = convert_pil_to_cv2(image)
     if img.ndim == 2:
@@ -59,12 +59,16 @@ def main(image):
     return image_output
 
 
+examples = [f"examples/example_{i+1}.png" for i in range(5)]
+css = ".output-image, .input-image, .image-preview {height: 480px !important} "
+
 gr.Interface(
-    main,
-    gr.inputs.Image(type="pil"),
-    "image",
+    fn=upscale,
+    inputs=gr.inputs.Image(type="pil"),
+    outputs="image",
+    examples=examples,
     title="Image Upscaling 🦆",
     allow_flagging="never",
-    css=".output-image, .input-image, .image-preview {height: 500px !important} ",
+    css=css,
 ).launch()
 

requirements.txt

@@ -1,3 +1,3 @@
 numpy
-onnxruntime
+onnxruntime-gpu
 opencv-python-headless