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"}}
+]