Update app.py

app.py

@@ -1,11 +1,10 @@
-import tensorflow as tf
 import gradio as gr
 import cv2
 import numpy as np
 from skimage import feature
 
 # Model path...
-edges2ferrets = './e10000_generator.hd5'
+edges2ferrets = './e299999_generator.h5'
 
 # Load the Pix2Pix generator model
 ferretMaker = tf.keras.models.load_model(edges2ferrets)
@@ -31,11 +30,11 @@ def transform_image(img, input_type):
         grey_img = cv2.cvtColor(img, cv2.COLOR_RGB2GRAY)
         processed_input = cv2.merge([grey_img, grey_img, grey_img])
     else:
-        raise ValueError("Invalid input_type. Choose 'Photo' or 'Drawing'.")
+        raise ValueError("Invalid input_type. Choose 'photo' or 'drawing'.")
 
     # Ferret-ing the input!
     processed_input = (processed_input - 127.5) / 127.5
-    result = ferretMaker(np.expand_dims(processed_input, 0), training=False)[0].numpy()
+    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)
@@ -45,7 +44,7 @@ iface = gr.Interface(
     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!",
+    title="Edges2Ferrets generation!",
     description="Choose between photo or drawing input to create a ferret!",
 )