Add harcoded keywords (#1)

- Add harcoded keys (1d351f53e3c5f53c9922a0da1c24446afcd42fd9)
- Nits (06c2073ef7c0bde7ffb8912b5924613cb844fe15)


Co-authored-by: Manon Blanco <[email protected]>

app.py

@@ -1,3 +1,35 @@
 import gradio as gr
 
-gr.load("models/artificialguybr/LogoRedmond-LogoLoraForSDXL-V2").launch()
+from diffusers import DiffusionPipeline
+
+
+KEYWORDS = [
+    "cute",
+    "small",
+    "cartoon",
+]
+
+PIPELINE = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-xl-base-1.0")
+PIPELINE.load_lora_weights("artificialguybr/LogoRedmond-LogoLoraForSDXL-V2")
+
+
+def predict(user_prompt: str):
+    prompt = ", ".join(KEYWORDS)
+    if user_prompt:
+        prompt += ", " + user_prompt
+
+    return PIPELINE(prompt).images[0]
+
+
+gradio_app = gr.Interface(
+    predict,
+    inputs=[
+        gr.Textbox(),
+    ],
+    outputs=[gr.Image()],
+    title="Cute Logo Creator",
+)
+
+
+if __name__ == "__main__":
+    gradio_app.launch()