Update app.py

app.py

@@ -1,19 +1,25 @@
-import gradio as gr
+# Installs
+!pip install gradio
+
+# Imports
 import tensorflow as tf
+import gradio as gr
 import cv2
 import numpy as np
+from PIL import Image
+import sys
+import os
 from skimage import feature
 
 # Model path...
-edges2ferrets = './e299999_generator.h5'
+edges2ferrets = './FINAL_EDGES2FERRETS.hd5'
 
-# Load the Pix2Pix generator model
 ferretMaker = tf.keras.models.load_model(edges2ferrets)
 
 def transform_image(img, input_type):
-    sigma = 1.7  # Placeholder value for sigma
+    sigma = 1.7
     if input_type == "Photo":
-        # Resize the input photo to 256x256
+        # Resize the input
         img = cv2.resize(img, (256, 256))
 
         # Convert the input photo to grayscale
@@ -24,7 +30,7 @@ def transform_image(img, input_type):
         processed_input = cv2.cvtColor(edges, cv2.COLOR_GRAY2RGB)
 
     elif input_type == "Drawing":
-        # Resize the input drawing to 256x256
+        # Resize the input
         img = cv2.resize(img, (256, 256))
 
         # Applies CV2 transformation to input drawing
@@ -40,6 +46,7 @@ def transform_image(img, input_type):
 
     return generated_ferret.astype(np.uint8)
 
+# Different canny logic is applied depending on the input the user selects. With photo, skimage is applied, and with drawing cv2 bw is applied.
 iface = gr.Interface(
     fn=transform_image,
     inputs=[gr.Image(type="numpy", image_mode="RGB"), gr.Radio(["Photo", "Drawing"], label="Select Input Type")],