touchups

app.py

@@ -57,6 +57,8 @@ def generate_text(prompt, extra=False, top_k=100, top_p=0.95, temperature=0.85,
     if model is None:
         return ["Try Again"] * 4
     
+    prompt = prompt.strip()
+    
     if not prompt.startswith("BRF:"):
         prompt = "BRF: " + prompt
 
@@ -75,6 +77,7 @@ def generate_text(prompt, extra=False, top_k=100, top_p=0.95, temperature=0.85,
             for output in outputs:
                 sample = tokenizer.decode(output, skip_special_tokens=True)
                 sample = format_prompt(sample, enhancers, inpspiration, negative_prompt)
+                print(f"Sample: {sample}")
                 samples.append(sample)
     except Exception as e:
         print(e)
@@ -86,10 +89,12 @@ if __name__ == "__main__":
         with gr.Row():
             gr.Markdown("""# Midjourney / Dalle 2 / Stable Diffusion Prompt Generator
     This is the 355M parameter model.  There is also a 7B parameter model that is much better but far slower (access coming soon).
-    Just enter a basic prompt and the fungineering model will use its wildest imagination to expand the prompt in detail.  You can then use this prompt to generate images with Midjourney, Dalle 2, Stable Diffusion, Bing Image Creator, or any other image generation model.  Treat this model more like a text-to-text model (simple prompt > complex prompt) rather than a generative model (prefix + word generation).  It is a generative model under the hood.""")
+    Just enter a basic prompt and the fungineering model will use its wildest imagination to expand the prompt in detail.  You can then use this prompt to generate images with Midjourney, Dalle 2, Stable Diffusion, Bing Image Creator, or any other image generation model.  Treat this model more like a text-to-text model (simple prompt > complex prompt) rather than a generative model (prefix + word generation).  It is a generative model under the hood.
+    ## TIP: Keep the base prompt short and simple.  The model will do the rest.
+    """)
         with gr.Row():
             with gr.Column():
-                base_prompt = gr.Textbox(lines=5, label="Base Prompt", placeholder="An astronaut in space", info="Enter a very simple prompt that will be fungineered into something exciting!")
+                base_prompt = gr.Textbox(lines=1, label="Base Prompt (Shorter is Better)", placeholder="An astronaut in space.", info="Enter a very simple prompt that will be fungineered into something exciting!")
                 extra = gr.Checkbox(value=True, label="Extra Fungineer Imagination", info="If checked, the model will be allowed to go wild with its imagination.")
                 with gr.Accordion("Advanced Generation Settings", open=False):
                     top_k = gr.Slider( minimum=10, maximum=1000, value=100, label="Top K", info="Top K sampling")