Update app.py

app.py

@@ -298,12 +298,13 @@ def stable_diffusion_compose(prompt, scale, steps):
 
 
 def compose(prompt, version, guidance_scale, steps):
-    if version == 'GLIDE':
-        return compose_language_descriptions(prompt, guidance_scale, steps)
-    elif version == 'Stable_Diffusion_1v_4':
-        return stable_diffusion_compose(prompt, guidance_scale, steps)
-    else:
-        return compose_clevr_objects(prompt, guidance_scale, steps)
+    with th.no_grad():
+        if version == 'GLIDE':
+            return compose_language_descriptions(prompt, guidance_scale, steps)
+        elif version == 'Stable_Diffusion_1v_4':
+            return stable_diffusion_compose(prompt, guidance_scale, steps)
+        else:
+            return compose_clevr_objects(prompt, guidance_scale, steps)
 
 
 examples_1 = 'a camel | a forest'
@@ -332,7 +333,7 @@ iface = gr.Interface(compose,
                      inputs=[
                          "text",
                          gr.Radio(['Stable_Diffusion_1v_4', 'GLIDE', 'CLEVR Objects'], type="value", label='version'),
-                         gr.Slider(2, 15),
+                         gr.Slider(2, 30),
                          gr.Slider(10, 200)
                      ],
                      outputs='image',