app.py

app.py

@@ -69,12 +69,18 @@ input = [gr.inputs.Image(type="filepath", label="Ground Truth - Real Image (jpg)
          gr.inputs.Image(type="filepath", label="Segementated image (png)"),
          gr.inputs.Image(type="filepath", label="corresponding bitmap image (bmp)")]
 
-facades_data = []
-data_dir = 'examples/'
-for idx, images in enumerate(os.listdir(data_dir)):
-  image = os.path.join(data_dir, images)
-  if os.path.isfile(image) and idx < 6:
-    facades_data.append(image)
+#facades_data = []
+#data_dir = 'examples/'
+#for idx, images in enumerate(os.listdir(data_dir)):
+#  image = os.path.join(data_dir, images)
+#  if os.path.isfile(image) and idx < 6:
+#    facades_data.append(image)
+
+examples = [["/content/facades_data/cmp_b0010.jpg", "/content/facades_data/cmp_b0010.png", "/content/facades_data/cmp_b0010.bmp"],
+            ["/content/facades_data/cmp_b0020.jpg", "/content/facades_data/cmp_b0020.png", "/content/facades_data/cmp_b0020.bmp"],
+            ["/content/facades_data/cmp_b0030.jpg", "/content/facades_data/cmp_b0030.png", "/content/facades_data/cmp_b0030.bmp"],
+            ["/content/facades_data/cmp_b0040.jpg", "/content/facades_data/cmp_b0040.png", "/content/facades_data/cmp_b0040.bmp"], 
+            ["/content/facades_data/cmp_b0050.jpg", "/content/facades_data/cmp_b0050.png", "/content/facades_data/cmp_b0050.bmp"]]
 
 # output
 output = [gr.outputs.Image(type="numpy", label="Generated - Conditioned Images")]
@@ -82,5 +88,5 @@ output = [gr.outputs.Image(type="numpy", label="Generated - Conditioned Images")
 title = "GauGAN For Conditional Image Generation"
 description = "Upload an Image or take one from examples to generate realistic images that are conditioned on cue images and segmentation maps"
 
-gr.Interface(fn=predict, inputs = input, outputs = output, examples=facades_data, allow_flagging=False, analytics_enabled=False,
+gr.Interface(fn=predict, inputs = input, outputs = output, examples=examples, allow_flagging=False, analytics_enabled=False,
   title=title, description=description, article="<center>Space By: <u><a href='https://github.com/robotjellyzone'><b>Kavya Bisht</b></a></u> \n Based on <a href='https://keras.io/examples/generative/gaugan/'><b>this notebook</b></a></center>").launch(enable_queue=True, debug=True)