Update app_ctrlx.py

app_ctrlx.py

@@ -160,14 +160,25 @@ description = """<div>
 
 @spaces.GPU
 def inference(
-    structure_image, appearance_image,
-    prompt, structure_prompt, appearance_prompt,
-    positive_prompt, negative_prompt,
-    guidance_scale, structure_guidance_scale, appearance_guidance_scale,
-    num_inference_steps, eta, seed,
-    width, height,
-    structure_schedule, appearance_schedule, use_advanced_config,
-    control_config,
+    structure_image,
+    appearance_image,
+    prompt,
+    structure_prompt,
+    appearance_prompt,
+    positive_prompt="high quality",
+    negative_prompt="ugly, blurry, dark, low res, unrealistic",
+    guidance_scale=5.0,
+    structure_guidance_scale=5.0,
+    appearance_guidance_scale=5.0,
+    num_inference_steps=28,
+    eta=1.0,
+    seed=42,
+    width=1024,
+    height=1024,
+    structure_schedule=0.6,
+    appearance_schedule=0.6,
+    use_advanced_config=False,
+    control_config="",
     progress=gr.Progress(track_tqdm=True)
 ):
     torch.manual_seed(seed)
@@ -308,111 +319,98 @@ with gr.Blocks(theme=gr.themes.Default(), css=css, title="Ctrl-X (SDXL v1.0)") a
             [
                 "assets/images/horse__point_cloud.jpg",
                 "assets/images/horse.jpg",
+                "a photo of a horse standing on grass",
                 "a 3D point cloud of a horse",
                 "",
-                "a photo of a horse standing on grass",
-                0.6, 0.6,
             ],
             [
                 "assets/images/cat__mesh.jpg",
                 "assets/images/tiger.jpg",
+                "a photo of a tiger standing on snow",
                 "a 3D mesh of a cat",
                 "",
-                "a photo of a tiger standing on snow",
-                0.6, 0.6,
             ],
             [
                 "assets/images/dog__sketch.jpg",
                 "assets/images/squirrel.jpg",
+                "a photo of a squirrel",
                 "a sketch of a dog",
                 "",
-                "a photo of a squirrel",
-                0.6, 0.6,
             ],
             [
                 "assets/images/living_room__seg.jpg",
                 "assets/images/van_gogh.jpg",
+                "a Van Gogh painting of a living room",
                 "a segmentation map of a living room",
                 "",
-                "a Van Gogh painting of a living room",
-                0.6, 0.6,
             ],
             [
                 "assets/images/bedroom__sketch.jpg",
                 "assets/images/living_room_modern.jpg",
                 "a sketch of a bedroom",
-                "",
                 "a photo of a modern bedroom during sunset",
-                0.6, 0.6,
+                "",
             ],
             [
                 "assets/images/running__pose.jpg",
                 "assets/images/man_park.jpg",
+                "a photo of a man running in a park",
                 "a pose image of a person running",
                 "",
-                "a photo of a man running in a park",
-                0.4, 0.6,
             ],
             [
                 "assets/images/fruit_bowl.jpg",
                 "assets/images/grapes.jpg",
+                "a photo of a bowl of grapes in the trees",
                 "a photo of a bowl of fruits",
                 "",
-                "a photo of a bowl of grapes in the trees",
-                0.6, 0.6,
             ],
             [
                 "assets/images/bear_avocado__spatext.jpg",
                 None,
+                "a realistic photo of a bear and an avocado in a forest",
                 "a segmentation map of a bear and an avocado",
                 "",
-                "a realistic photo of a bear and an avocado in a forest",
-                0.6, 0.6,
             ],
             [
                 "assets/images/cat__point_cloud.jpg",
                 None,
+                "an embroidery of a white cat sitting on a rock under the night sky",
                 "a 3D point cloud of a cat",
                 "",
-                "an embroidery of a white cat sitting on a rock under the night sky",
-                0.6, 0.6,
             ],
             [
                 "assets/images/library__mesh.jpg",
                 None,
+                "a Polaroid photo of an old library, sunlight streaming in",
                 "a 3D mesh of a library",
                 "",
-                "a Polaroid photo of an old library, sunlight streaming in",
-                0.6, 0.6,
             ],
             [
                 "assets/images/knight__humanoid.jpg",
                 None,
+                "a photo of a medieval soldier standing on a barren field, raining",
                 "a 3D model of a person holding a sword and shield",
                 "",
-                "a photo of a medieval soldier standing on a barren field, raining",
-                0.6, 0.6,
             ],
             [
                 "assets/images/person__mesh.jpg",
                 None,
+                "a photo of a Karate man performing in a cyberpunk city at night",
                 "a 3D mesh of a person",
                 "",
-                "a photo of a Karate man performing in a cyberpunk city at night",
-                0.5, 0.6,
             ],
         ],
         [
             structure_image,
             appearance_image,
+            prompt,
             structure_prompt,
             appearance_prompt,
-            prompt,
-            structure_schedule,
-            appearance_schedule,
         ],
         examples_per_page=50,
-        cache_examples="lazy"
+        cache_examples="lazy",
+        outputs=[result, result_refiner, structure_recon, appearance_recon]
     )
 
 app.launch(debug=False, share=False)