Create app.py

app.py

@@ -0,0 +1,34 @@
+#@title Text2Art Gradio UI
+import gradio as gr
+import torch
+import clipit
+
+# Define the main function
+def generate(prompt, quality, style, aspect):
+    torch.cuda.empty_cache()
+    clipit.reset_settings()
+    
+    use_pixeldraw = (style == 'pixel art')
+    use_clipdraw = (style == 'painting')
+    clipit.add_settings(prompts=prompt,
+                        aspect=aspect,
+                        quality=quality,
+                        use_pixeldraw=use_pixeldraw,
+                        use_clipdraw=use_clipdraw,
+                        make_video=True)
+  
+    settings = clipit.apply_settings()
+    clipit.do_init(settings)
+    clipit.do_run(settings)
+
+    return 'output.png', 'output.mp4'
+
+# Create the UI
+prompt = gr.inputs.Textbox(default="Underwater city", label="Text Prompt")
+quality = gr.inputs.Radio(choices=['draft', 'normal', 'better'], label="Quality")
+style = gr.inputs.Radio(choices=['image', 'painting','pixel art'], label="Type")
+aspect = gr.inputs.Radio(choices=['square', 'widescreen','portrait'], label="Size")
+
+# Launch the demo
+iface = gr.Interface(generate, inputs=[prompt, quality, style, aspect], outputs=['image', 'video'], enable_queue=True, live=False)
+iface.launch(debug=True)