add examples

app.py

@@ -114,7 +114,6 @@ class InrenceTextVAR(nn.Module):
 
 
 if __name__ == '__main__':
-
     # Initialize the model
     checkpoint = 'VARtext_v1.pth'  # Replace with your actual checkpoint path
     device = 'cpu' if not torch.cuda.is_available() else 'cuda'
@@ -123,6 +122,7 @@ if __name__ == '__main__':
     model.load_state_dict(state_dict)
     model.to(device)
 
+
     def generate_image_gradio(text, beta=1.0, seed=None, more_smooth=False, top_k=0, top_p=0.9):
         print(f"Generating image for text: {text}\n"
               f"beta: {beta}\n"
@@ -133,8 +133,9 @@ if __name__ == '__main__':
         image = model.generate_image(text, beta=beta, seed=seed, more_smooth=more_smooth, top_k=int(top_k), top_p=top_p)
         return image
 
+
     with gr.Blocks() as demo:
-        gr.Markdown("# Text to Image/Video Generator")
+        gr.Markdown("# PopYou2-VAR")
         with gr.Tab("Generate Image"):
             text_input = gr.Textbox(label="Input Text")
             beta_input = gr.Slider(label="Beta", minimum=0.0, maximum=2.5, step=0.05, value=1.0)
@@ -150,4 +151,16 @@ if __name__ == '__main__':
                 outputs=image_output
             )
 
-    demo.launch()
+        gr.Markdown("### Examples")
+        with gr.Row():
+            example1_text = gr.Textbox(label="Example 1", value="a funko pop figure of a yellow robot tom cruise with headphones on a white background", interactive=False)
+            example1_image = gr.Image(label="Generated Image 1", value="examples/tom_cruise_robot.png")  # Replace with the actual path
+        with gr.Row():
+            example2_text = gr.Textbox(label="Example 2", value="a funko pop figure of a alien Scarlett Johansson holding a shield on a white background", interactive=False)
+            example2_image = gr.Image(label="Generated Image 2", value="examples/alien_Scarlett_Johansson.png")  # Replace with the actual path
+        with gr.Row():
+            example3_text = gr.Textbox(label="Example 3", value="a funko pop figure of a woman with a hat and a pink long hair and blue dress on a white background", interactive=False)
+            example3_image = gr.Image(label="Generated Image 3", value="examples/woman_pink.png")  # Replace with the actual path
+
+    demo.launch()
+