Add output examples, use EulerA sampler and reduce inference steps to 20 to reduce CPU runtime from 20 minutes.

app.py

@@ -21,7 +21,7 @@ import random
 from transformers import GPT2TokenizerFast
 
 import torch
-from diffusers import DiffusionPipeline
+from diffusers import DiffusionPipeline, EulerAncestralDiscreteScheduler
 
 gpu = False
 
@@ -30,10 +30,14 @@ AUTH_TOKEN = os.environ.get('AUTH_TOKEN')
 
 if gpu:
     pipeline = DiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5", custom_pipeline="stable_diffusion_mega", torch_dtype=torch.float16, revision="fp16", use_auth_token=AUTH_TOKEN)
+    scheduler = EulerAncestralDiscreteScheduler.from_config(pipeline.scheduler.config)
+    pipeline.scheduler = scheduler
     pipeline.to("cuda")
 else:
     pipeline = DiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5",
                                                  custom_pipeline="stable_diffusion_mega", use_auth_token=AUTH_TOKEN)
+    scheduler = EulerAncestralDiscreteScheduler.from_config(pipeline.scheduler.config)
+    pipeline.scheduler = scheduler
 
 # Huggingface Spaces have 16GB RAM and 8 CPU cores
 # See https://huggingface.co/docs/hub/spaces-overview#hardware-resources
@@ -105,7 +109,7 @@ def run(name):
     prompt_template = f"fantasy illustration of a {card_type} {name}, by Greg Rutkowski"
     print(f"GENERATING IMAGE FOR {prompt_template}")
     # Regarding sizing see https://huggingface.co/blog/stable_diffusion#:~:text=When%20choosing%20image%20sizes%2C%20we%20advise%20the%20following%3A
-    images = pipeline.text2img(prompt_template, width=512, height=368).images
+    images = pipeline.text2img(prompt_template, width=512, height=368, num_inference_steps=20).images
     card_image = None
     for image in images:
         save_name = get_savename('card_images', name, 'png')
@@ -258,12 +262,12 @@ def html_to_png(card_name, html):
 app_description = (
     """
     # Create your own Magic: The Gathering cards!
-    Enter a name, click Submit, and wait for about 20 seconds to see the result.
+    Enter a name, click Submit, it may take up to 10 minutes to run on the free CPU only instance.
     """).strip()
 input_box = gr.Textbox(label="Enter a card name", placeholder="Firebolt")
-rendered_card = gr.Image(label="Card", type='pil')
-output_text_box = gr.Textbox(label="Card Text")
-output_card_image = gr.Image(label="Card Image", type='pil')
+rendered_card = gr.Image(label="Card", type='pil', value="examples/card.png")
+output_text_box = gr.Textbox(label="Card Text", value=pathlib.Path("examples/text.txt").read_text('utf-8'))
+output_card_image = gr.Image(label="Card Image", type='pil', value="examples/image.png")
 output_card_html = gr.HTML(label="Card HTML", visible=False, show_label=False)
 x = gr.components.Textbox()
 iface = gr.Interface(title="MagicGen", theme="default", description=app_description, fn=run, inputs=[input_box],

examples/text.txt

@@ -0,0 +1,10 @@
+Name: Firebolt
+ManaCost: {1}{R}
+Type: Instant
+Rarity: common
+Text: Firebolt deals 2 damage to target creature and you gain 2 life.
+Draw a card at the beginning of the next end step.
+FlavorText:
+Power:
+Toughness:
+Color: ['R']