Update app.py

app.py

@@ -306,33 +306,33 @@ def compose(prompt, version, guidance_scale, steps):
             model_up.to(device)
             return compose_language_descriptions(prompt, guidance_scale, steps)
         elif version == 'Stable_Diffusion_1v_4':
+            clevr_model.to(cpu)
             model.to(cpu)
             model_up.to(cpu)
-            clevr_model.to(cpu)
             pipe.to(device)
             return stable_diffusion_compose(prompt, guidance_scale, steps)
         else:
+            pipe.to(cpu)
             model.to(cpu)
             model_up.to(cpu)
-            pipe.to(cpu)
             clevr_model.to(device)
             return compose_clevr_objects(prompt, guidance_scale, steps)
 
 
 examples_1 = 'a camel | a forest'
-examples_2 = 'A cloudy blue sky  | A mountain in the horizon | Cherry Blossoms in front of the mountain'
+examples_2 = 'A blue sky  | A mountain in the horizon | Cherry Blossoms in front of the mountain'
 examples_3 = '0.1, 0.5 | 0.3, 0.5 | 0.5, 0.5 | 0.7, 0.5 | 0.9, 0.5'
-examples_4 = 'a river leading into a mountain | red trees on the side'
-examples_5 = 'a white church on a hill | birds flying around the church'
-examples_6 = 'a boat in a desert | a pink sky'
-examples_7 = 'mountains in the background |  a blue sky | cows on a pasture'
+examples_4 = 'red trees | a blue house'
+examples_5 = 'a white church | lightning in the background'
+examples_6 = 'a camel | arctic'
+examples_7 = 'A lake | A mountain  | Cherry Blossoms next to the lake'
 examples = [
-            [examples_7, 'Stable_Diffusion_1v_4', 30, 50],
-            [examples_4, 'Stable_Diffusion_1v_4', 30, 50],
-            [examples_5, 'Stable_Diffusion_1v_4', 30, 50],
-            [examples_6, 'Stable_Diffusion_1v_4', 30, 50],
-            [examples_1, 'GLIDE', 30, 100],
-            [examples_2, 'GLIDE', 30, 100],
+            [examples_7, 'Stable_Diffusion_1v_4', 15, 50],
+            [examples_5, 'Stable_Diffusion_1v_4', 15, 50],
+            [examples_4, 'Stable_Diffusion_1v_4', 20, 50],
+            [examples_6, 'Stable_Diffusion_1v_4', 15, 50],
+            [examples_1, 'GLIDE', 15, 100],
+            [examples_2, 'GLIDE', 15, 100],
             [examples_3, 'CLEVR Objects', 10, 100]
 ]