rudalle

app.py

@@ -110,6 +110,25 @@ def desc_to_image(desc):
   img=iface(desc, steps, width, height, images, diversity)[0]
   return img
 
+def desc_to_image_dalle(desc):
+  print("*****Inside desc_to_image")
+  desc = " ".join(desc.split('\n'))
+  desc = desc + ", character art, concept art, artstation"
+  steps, width, height, images, diversity = '50','256','256','1',15
+  #iface = gr.Interface.load("huggingface/flax-community/dalle-mini")#this isn't a real interface
+  iface = gr.Interface.load("spaces/multimodalart/rudalle")
+  print("about to die",iface,dir(iface))
+
+  prompt = re.sub(r'[^a-zA-Z0-9 ,.]', '', desc)
+  print("about to die",prompt)
+
+  model='Realism'
+  aspect_ratio = 'Square'
+
+
+  img=iface(desc,model,aspect_ratio)[0]
+  return img
+
 demo = gr.Blocks()
 
 with demo:
@@ -124,7 +143,8 @@ with demo:
   with gr.Row():
     b0 = gr.Button("Randomize name,race and class")
     b1 = gr.Button("Generate NPC Description")
-    b2 = gr.Button("Generate Portrait")
+    b2 = gr.Button("Generate Portrait (latent diffusion)")
+    b3 = gr.Button("Generate Portrait (rudalle)")
   
   with gr.Row():  
     input_name = gr.Textbox(label="name",placeholder="Drizzt")
@@ -139,6 +159,7 @@ with demo:
   b0.click(npc_randomize,inputs=[],outputs=[input_name,input_race,input_class,input_pronoun])
   b1.click(npc_generate, inputs=[ input_name,input_race,input_class,input_pronoun], outputs=desc_txt)
   b2.click(desc_to_image, desc_txt, output_image)
+  b3.click(desc_to_image_dalle, desc_txt, output_image)
   #examples=examples
 
 demo.launch(enable_queue=True, debug=True)