initial gradio app for hf space

app.py

@@ -1,7 +1,22 @@
 import gradio as gr
+from transformers import pipeline
 
-def greet(name):
-    return "Hello " + name + "!!"
+generator = pipeline(model="cactusfriend/nightmare-invokeai-prompts",
+                    tokenizer="cactusfriend/nightmare-invokeai-prompts",
+                    task="text-generation")
 
-iface = gr.Interface(fn=greet, inputs="text", outputs="text")
+def makePrompts(prompt: str, temp: float = 1.8):
+    outputs = generator(prompt, max_new_tokens=100,
+                        temperature=temp, do_sample=True,
+                        top_p=0.9, top_k=40, num_return_sequences=5)
+    items = set([i['generated_text'] for i in outputs])
+    return "\n --- \n".join(items)
+
+
+iface = gr.Interface(fn=makePrompts,
+                     inputs=[gr.Textbox(lines=1, label="Prompt", info="the start of the prompts", placeholder="a photograph of"),
+                             gr.Slider(0.5, 3.0, value=1.8, label="Temperature", info="higher = crazier")],
+                     outputs=gr.Textbox(lines=10, label="Five prompts"),
+                     title="Nightmare InvokeAI Prompts",
+                     allow_flagging="never")
 iface.launch()

requirements.txt

@@ -0,0 +1,2 @@
+transformers
+torch