Create app.py

app.py

@@ -0,0 +1,29 @@
+# Model path...
+edges2ferrets = '/content/drive/MyDrive/DMLAP_FINAL/models/edges2ferrets_ver2.2/e299999_generator.h5'
+
+# Load the Pix2Pix generator model
+ferretMaker = tf.keras.models.load_model(edges2ferrets)
+
+
+def transform_image(img):
+    # Applies CV2 transformation to input doodle
+    grey_img = cv2.cvtColor(img, cv2.COLOR_RGB2GRAY)
+    processed_input = cv2.merge([grey_img, grey_img, grey_img])
+
+    # Ferret-ing the input!
+    processed_input = (processed_input - 127.5) / 127.5
+    result = ferretMaker(np.expand_dims(processed_input, 0), training=True)[0].numpy()
+    generated_ferret = (result * 0.5 + 0.5) * 255.0
+
+    return generated_ferret.astype(np.uint8)
+
+iface = gr.Interface(
+    fn=transform_image,
+    inputs=gr.Image(type="numpy", image_mode="RGB"),
+    outputs=gr.Image(type="numpy", image_mode="RGB"),
+    live=True,
+    title="Edges2Ferrets generation!",
+    description="Upload a doodle to create a ferret!",
+)
+
+iface.launch(share=True)