revert to latentdiffusion model, parameter tweaks

app.py

@@ -14,7 +14,7 @@ from huggingface_hub import snapshot_download
 
 with warnings.catch_warnings():
   warnings.simplefilter('ignore')
-  image_gen = gr.Interface.load("spaces/multimodalart/diffusion")
+  image_gen = gr.Interface.load("spaces/multimodalart/latentdiffusion")
 
   os.system("git clone https://github.com/google-research/frame-interpolation")
   sys.path.append("frame-interpolation")
@@ -38,9 +38,10 @@ def generate_story(choice, input_text):
     return generated_text
 
 
-def generate_images(text, steps=40, num_images=2, weight=10):
+def generate_images(text, width=256, height=256, steps=50, num_images=2,
+                    diversity=3):
 
-    image_bytes = image_gen(text, steps, num_images, weight, False)
+    image_bytes = image_gen(text, steps, width, height, num_images, diversity)
     
     # Algo from spaces/Gradio-Blocks/latent_gpt2_story/blob/main/app.py
     generated_images = []
@@ -68,7 +69,6 @@ def generate_interpolation(text, fps=5):
     mediapy.write_video("out.mp4", vid, fps=fps)
     return "out.mp4"
 
-
 demo = gr.Blocks()
 
 with demo: