Update app.py

app.py

@@ -1,11 +1,46 @@
+import tensorflow as tf 
+import pathlib
 import gradio as gr
+import matplotlib.pyplot as plt 
+from huggingface_hub import from_pretrained_keras
+import numpy as np
+
+# Normalizing the images to [-1, 1]
+def normalize_test(input_image):
+  input_image = tf.cast(input_image, tf.float32)
+  input_image = (input_image / 127.5) - 1
+  return input_image
+  
+def resize(input_image, height, width):
+  input_image = tf.image.resize(input_image, [height, width],
+                                method=tf.image.ResizeMethod.NEAREST_NEIGHBOR)
+  return input_image
+  
+def load_image_infer(image_file):
+  input_image = resize(image_file, 256, 256)
+  input_image = normalize_test(input_image)
+
+  return input_image
+  
+def generate_images(test_input):
+  test_input = load_image_infer(test_input)
+  prediction = generator(np.expand_dims(test_input, axis=0), training=True)
+  fig = plt.figure(figsize=(128, 128))
+  title = ['Predicted Image']
+
+  plt.title('Predicted Image')
+  # Getting the pixel values in the [0, 1] range to plot.
+  plt.imshow(prediction[0,:,:,:] * 0.5 + 0.5)
+  plt.axis('off')
+  return fig
+  
+  
+generator = from_pretrained_keras("keras-io/pix2pix-generator")
 
-def sketch_to_shoes(img):
-  # Implement sketch recognition model here...
-  # Return labels and confidences as dictionary
-   return img
 
 img = gr.inputs.Image(shape=(256,256))
 plot = gr.outputs.Image(type="plot")
 
-iface = gr.Interface(fn=sketch_to_shoes, inputs="sketchpad", title = "Pix2Pix Sketch to Shoes", outputs=plot).launch()
+description = "Conditional GAN model that translates image-to-image."
+gr.Interface(generate_images, inputs = img, outputs = plot, 
+title = "Pix2Pix Shoes Reconstructor", description = description, examples = [["./img.png"]]).launch()