Update app.py

app.py

@@ -13,7 +13,9 @@ ip_ckpt = hf_hub_download(repo_id="CiaraRowles/stylecodes", filename="stylecodes
 
 device = "cuda"
 
-
+example_path = os.path.join(os.path.dirname(__file__), 'test_images')
+list = os.listdir(example_path)
+list_path = [os.path.join(example_path,img) for img in list]
 
 cv2.setNumThreads(1)
 
@@ -47,6 +49,10 @@ def swap_to_gallery(images):
 
 def remove_back_to_files():
     return gr.update(visible=False), gr.update(visible=False), gr.update(visible=True)
+
+
+
+    
 css = '''
 h1{margin-bottom: 0 !important}
 '''
@@ -62,8 +68,9 @@ with gr.Blocks(css=css) as demo:
             uploaded_files = gr.Gallery(label="Your image", visible=False, columns=5, rows=1, height=125)
             with gr.Column(visible=False) as clear_button:
                 remove_and_reupload = gr.ClearButton(value="Remove and upload new ones", components=files, size="sm")
-            stylecode_gen = gr.Button("Make stylecode")
             stylecode = gr.Textbox(label="stylecode")
+            stylecode_gen = gr.Button("Make stylecode")
+            
             
             prompt = gr.Textbox(label="Prompt",
                 info="Try something like 'a photo of a man/woman/person'",
@@ -80,7 +87,11 @@ with gr.Blocks(css=css) as demo:
             submit = gr.Button("Use stylecode")
         with gr.Column():
             gallery = gr.Gallery(label="Generated Images")
-
+            examples = gr.Examples(
+                inputs=uploaded_files,
+                examples_per_page=4,
+                examples=list_path
+            )
         stylecode_gen.click(fn=make_stylecode_gui,inputs=[files],outputs=stylecode)
         
         submit.click(fn=generate_image,