Update app.py

app.py

@@ -14,14 +14,14 @@ from previewer.modules import Previewer
 os.environ['TOKENIZERS_PARALLELISM'] = 'false'
 
 DESCRIPTION = "# Stable Cascade"
-DESCRIPTION += "\n<p style=\"text-align: center\">Unofficial demo Modified by <a href='https://www.youtube.com/@Aitrepreneur/' target='_blank'>Aitrepreneur</a> for <a href='https://huggingface.co/stabilityai/stable-cascade' target='_blank'>Stable Cascade</a>, a new high resolution text-to-image model by Stability AI, built on the Würstchen architecture - <a href='https://huggingface.co/stabilityai/stable-cascade/blob/main/LICENSE' target='_blank'>non-commercial research license</a></p>"
+DESCRIPTION += "\n<p style=\"text-align: center\">Unofficial V2 demo Modified by <a href='https://www.youtube.com/@Aitrepreneur/' target='_blank'>Aitrepreneur</a> for <a href='https://huggingface.co/stabilityai/stable-cascade' target='_blank'>Stable Cascade</a>, a new high resolution text-to-image model by Stability AI, built on the Würstchen architecture - <a href='https://huggingface.co/stabilityai/stable-cascade/blob/main/LICENSE' target='_blank'>non-commercial research license</a></p>"
 
 MAX_SEED = np.iinfo(np.int32).max
 CACHE_EXAMPLES = False
 MAX_IMAGE_SIZE = int(os.getenv("MAX_IMAGE_SIZE", "1536"))
 USE_TORCH_COMPILE = False
 ENABLE_CPU_OFFLOAD = os.getenv("ENABLE_CPU_OFFLOAD") == "1"
-PREVIEW_IMAGES = True
+PREVIEW_IMAGES = False
 
 parser = argparse.ArgumentParser(description='Gradio App Control')
 parser.add_argument('--share', action='store_true', help='Create a public shareable URL')
@@ -91,7 +91,7 @@ def generate(
     num_images_per_prompt: int = 2,
 #    profile: gr.OAuthProfile | None = None,
 ) -> PIL.Image.Image:
-    previewer.eval().requires_grad_(False).to(device).to(dtype)
+    #previewer.eval().requires_grad_(False).to(device).to(dtype)
     prior_pipeline.to(device)
     decoder_pipeline.to(device)
     
@@ -106,8 +106,8 @@ def generate(
         guidance_scale=prior_guidance_scale,
         num_images_per_prompt=num_images_per_prompt,
         generator=generator,
-        callback=callback_prior,
-        callback_steps=callback_steps
+        #callback=callback_prior,
+        #callback_steps=callback_steps
     )
 
     if PREVIEW_IMAGES:
@@ -270,4 +270,4 @@ if __name__ == "__main__":
         'share': args.share,
         'server_port' : args.server_port,
     }
-    demo_with_history.launch(**launch_args)
+    demo_with_history.launch(**launch_args)