example test

app.py

@@ -4,6 +4,8 @@ import gradio as gr
 import transformers
 import os
 import re
+import json
+
 
 device = "cpu"
 is_peft = False
@@ -83,10 +85,10 @@ Just enter a basic prompt and the fungineering model will use its wildest imagin
 
         with gr.Column():
             outputs = [
-                gr.Textbox(lines=5, label="Fungineered Text 1"),
-                gr.Textbox(lines=5, label="Fungineered Text 2"),
-                gr.Textbox(lines=5, label="Fungineered Text 3"),
-                gr.Textbox(lines=5, label="Fungineered Text 4"),
+                gr.Textbox(lines=3, label="Fungineered Text 1"),
+                gr.Textbox(lines=3, label="Fungineered Text 2"),
+                gr.Textbox(lines=3, label="Fungineered Text 3"),
+                gr.Textbox(lines=3, label="Fungineered Text 4"),
             ]
 
     inputs = [base_prompt, extra, top_k, top_p, temperature, enh, insp, neg]
@@ -95,5 +97,25 @@ Just enter a basic prompt and the fungineering model will use its wildest imagin
     submit = gr.Button(label="Fungineer",variant="primary")
     submit.click(generate_text, inputs=inputs, outputs=outputs)
 
+    examples = []
+    with open("examples.json") as f:
+        examples = json.load(f)
+
+    for i, example in enumerate(examples):
+        with gr.Tab(f"Example {i+1}"):
+            with gr.Row():
+                with gr.Column():
+                    gr.Markdown(f"### Base Prompt")
+                    gr.Image(value=f"{example['base']['src']}")
+                    gr.Markdown(f"**{example['base']['prompt']}**")
+                with gr.Column():
+                    gr.Markdown(f"### 355M Prompt Fungineered")
+                    gr.Image(value=f"{example['355M']['src']}")
+                    gr.Markdown(f"**{example['355M']['prompt']}**")
+                with gr.Column():
+                    gr.Markdown(f"### 7B Prompt Fungineered")
+                    gr.Markdown(f"**Coming Soon!**")
+
+
 fungineer.launch(enable_queue=True)
 

examples.json

@@ -0,0 +1,3 @@
+[
+    {"base":{"prompt":"basketball", "src":"static/basketball_base.png"}, "355M":{"prompt":"An imaginative image of a basketball player leaping into the air as they strike the ball. The player, dressed in an off-the-shoulder outfit, is adorned with a dazzling array of colorful and metallic studs, displaying their incredible speed and power. Their body is draped in a dazzling array of high-tech equipment, including a power rack, a motion-sensitive visor, and a ball gown. The background features a breathtaking cityscape filled with people in colorful attire, such as colorful sneakers, and a dramatic cityscape filled with skyscrapers. wide-angle lens, wide-angle lens, high contras","src":"static/basketball_355M.png"}}
+]