from pathlib import Path import gradio as gr ANIMALS = [animal.strip() for animal in Path("animals.txt").read_text().splitlines()] KEYWORDS = [ "cute", "small", "cartoon", ] MODEL = gr.Interface.load( "models/artificialguybr/LogoRedmond-LogoLoraForSDXL-V2", live=False, preprocess=True, postprocess=False, ) def predict(user_prompt: str): prompt = ", ".join(KEYWORDS) if user_prompt: prompt += ", " + user_prompt.lower() return MODEL(prompt) select_animal_tab = gr.Interface( predict, inputs=[ gr.Dropdown( choices=ANIMALS, value="Cat", filterable=True, label="Select an animal" ) ], outputs=[gr.Image(label="Your super cute animal logo 🥺", show_label=True)], allow_flagging="never", ) free_input_tab = gr.Interface( predict, inputs=[ gr.Textbox( placeholder="Enter your corporate keywords", label="Generate your Teklia logo", ) ], outputs=[gr.Image(label="Your super cute corporate logo 🥺", show_label=True)], allow_flagging="never", ) qte_app = gr.TabbedInterface( [select_animal_tab, free_input_tab], tab_names=["Cuteness overload", "TekQ-te"], title="Q-te logo creator", ) if __name__ == "__main__": qte_app.launch()