Update app.py

app.py

@@ -2,17 +2,16 @@ import tensorflow as tf
 import gradio as gr
 import cv2
 import numpy as np
-from PIL import Image
 from skimage import feature
 
 # Model path...
-edges2ferrets = './e299999_generator.h5'
+edges2ferrets = './edges2ferrets_model.h5'
 
 # 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.0  # Placeholder value for sigma
     if input_type == "photo":
         # Resize the input photo to 256x256
         img = cv2.resize(img, (256, 256))
@@ -43,7 +42,7 @@ def transform_image(img, input_type):
 
 iface = gr.Interface(
     fn=transform_image,
-    inputs=[gr.Image(type="numpy", image_mode="RGB"), gr.Radio(["Drawing","Photo"], label="Select Input Type")],
+    inputs=[gr.Image(type="numpy", image_mode="RGB"), gr.Radio(["Photo", "Drawing"], label="Select Input Type")],
     outputs=gr.Image(type="numpy", image_mode="RGB"),
     live=True,
     title="Edges2Ferrets generation!",