Meged minecraft and regular upscaler playground, added x2 models.

app.py

@@ -39,8 +39,8 @@ def convert_pil_to_cv2(image):
     return open_cv_image
 
 
-def upscale(image):
-    model_path = "models/model.ort"
+def upscale(image, model):
+    model_path = f"models/{model}.ort"
     img = convert_pil_to_cv2(image)
     if img.ndim == 2:
         img = cv2.cvtColor(img, cv2.COLOR_GRAY2BGR)
@@ -62,16 +62,29 @@ def upscale(image):
     return image_output
 
 
-examples = [f"examples/example_{i+1}.png" for i in range(5)]
+examples = [[f"examples/example_{i+1}.png", "modelx4"] for i in range(5)]
+examples += [
+    [f"minecraft_examples/minecraft-{j+1}.png", "minecraft_modelx4"] for j in range(5)
+]
 css = ".output-image, .input-image, .image-preview {height: 480px !important} "
+model_choices = ["modelx2", "modelx4", "minecraft_modelx4"]
 
 gr.Interface(
     fn=upscale,
-    inputs=gr.inputs.Image(type="pil"),
+    inputs=[
+        gr.inputs.Image(type="pil"),
+        gr.inputs.Radio(
+            model_choices,
+            type="value",
+            default=None,
+            label="Choose a Model",
+            optional=False,
+        ),
+    ],
     outputs="image",
     examples=examples,
+    examples_per_page=5,
     title="Image Upscaling 🦆",
     allow_flagging="never",
     css=css,
-).launch()
-
+).launch(debug=True)