add examples

gradio_interface.py

@@ -1,6 +1,12 @@
 import gradio as gr
 from PIL import Image
 
+examples = [
+ [Image.open("examples/in0.jpg"), Image.open("examples/out0.webp")],
+ [Image.open("examples/in1.webp"), Image.open("examples/out1.png")],
+ [Image.open("examples/in2.jpg"), Image.open("examples/out2.png")],
+ [Image.open("examples/in3.jpg"), Image.open("examples/out3.png")],
+]
 
 def create_gradio_interface(process_and_generate):
  def gradio_process_and_generate(input_image, prompt, num_images, cfg_weight):
@@ -15,6 +21,7 @@ Here, by feeding the model an image and then asking it to generate that same ima
  with gr.Blocks() as demo:
  gr.Markdown("# How Janus-1.3B sees itself")
 
+ dummy = gr.Image(type="filepath", label="Generated Image", visible=False)
  with gr.Row():
  input_image = gr.Image(type="filepath", label="Input Image")
  output_images = gr.Gallery(label="Generated Images", columns=2, rows=2)
@@ -28,6 +35,11 @@ Here, by feeding the model an image and then asking it to generate that same ima
  inputs=[input_image, prompt, num_images, cfg_weight],
  outputs=output_images
  )
+ gr.Examples(
+ examples=examples,
+ inputs=[input_image, dummy]
+ )
+ 
  gr.Markdown(explanation)
 
  return demo