metadata + requirements + layout

README.md

@@ -1,13 +1,16 @@
 ---
 title: CHARRED
 emoji: 📈
-colorFrom: gray
-colorTo: indigo
+colorFrom: red
+colorTo: navy
 sdk: gradio
-sdk_version: 3.27.0
+python_version: 3.10.6
+sdk_version: 3.28.0
 app_file: app.py
 pinned: true
 license: cc-by-sa-4.0
+models: [ "character-aware-diffusion/charred", "google/byt5-base", flax/stable-diffusion-2-1" ]
+datasets: [ "laion/laion-high-resolution" ]
 tags:
 - jax-diffusers-event
 ---

app.py

@@ -1,34 +1,26 @@
 import gradio as gr
-title = "Character aware stable difusion"
-description = "This is a demo of a character aware diffusion model"
-def infer_sd(prompt, negative_prompt, image):
-    # your inference function for stable diffusion control 
-    return None
 
-def infer_charr_sd(prompt, negative_prompt, image):
+title = "Character-Aware Stable Diffusion (CHARRED)"
+
+description = "This is a demo of the CHARRED character-aware stable diffusion model for you to enjoy at your own leisure, risk and peril"
+
+def infer_charred(prompt):
     # your inference function for charr stable difusion control 
     return None
 
 with gr.Blocks(theme='gradio/soft') as demo:
-    gr.Markdown("## Character aware stable difusion")
+ 
+    gr.Markdown("# Character aware stable difusion")
 
-    with gr.Tab("Character-Aware Difusion "):
+    with gr.Tab("Journal"):
+        gr.Markdown("## On How Four Crazy Fellow Embarked on Training a U-Net from Scratch in Five Days and Almost Died in the End.")
+
+    with gr.Tab("☢️ DEMO ☢️"):
         prompt_input_charr = gr.Textbox(label="Prompt")
-        negative_prompt_charr = gr.Textbox(label="Negative Prompt")
-        charr_input = gr.Image(label="Input Image")
         charr_output = gr.Image(label="Output Image")
         submit_btn = gr.Button(value = "Submit")
-        charr_inputs = [prompt_input_charr, negative_prompt_charr, charr_input]
-        submit_btn.click(fn=infer_charr_sd, inputs=charr_inputs, outputs=[charr_output])
-        
-    with gr.Tab("Stable Diffusion"):
-        prompt_input_seg = gr.Textbox(label="Prompt")
-        negative_prompt_seg = gr.Textbox(label="Negative Prompt")
-        sd_input = gr.Image(label="Input Image")
-        sd_output = gr.Image(label="Output Image")
-        submit_btn = gr.Button(value = "Submit")
-        sd_inputs = [prompt_input_seg, negative_prompt_seg, sd_input]
-        submit_btn.click(fn=infer_sd, inputs=sd_inputs, outputs=[sd_output])
+        charr_inputs = [prompt_input_charr]
+        submit_btn.click(fn=infer_charred, inputs=charr_inputs, outputs=[charr_output])
     #examples = [["postage stamp from california", "low quality", "charr_output.png", "charr_output.png" ]]
     #gr.Examples(fn = infer_sd, inputs = ["text", "text", "image", "image"], examples=examples, cache_examples=True)
 

requirements.txt

@@ -0,0 +1,13 @@
+-f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html
+jax[cuda12_pip]
+flax
+optax
+diffusers
+transformers
+datasets
+--extra-index-url https://download.pytorch.org/whl/cpu
+torch
+torchvision
+wandb
+Pillow
+gradio