Upload app.py

app.py

@@ -11,7 +11,7 @@ def apply_patches():
     
     # 1. Fix tqdm Jupyter/Thread Error
     try:
-        import tqdm
+        from tqdm.auto import tqdm
         if not hasattr(tqdm, '_is_patched'):
             import tqdm.notebook
             import tqdm.std
@@ -121,8 +121,8 @@ def custom_sampling_loop(prompt, pipe, guidance_scale=7.5, contrast_scale=0.0, c
     latents = torch.randn((num_images, pipe.unet.config.in_channels, 512 // 8, 512 // 8), generator=generator, device=device, dtype=dtype)
     pipe.scheduler.set_timesteps(num_inference_steps)
     latents = latents * pipe.scheduler.init_noise_sigma
-    import tqdm
-    for t in tqdm.auto.tqdm(pipe.scheduler.timesteps):
+    from tqdm.auto import tqdm
+    for t in tqdm(pipe.scheduler.timesteps):
         latent_model_input = torch.cat([latents] * 2)
         latent_model_input = pipe.scheduler.scale_model_input(latent_model_input, t)
         with torch.no_grad():
@@ -284,7 +284,8 @@ def generate_all_styles(prompt, seed1, seed2, seed3, seed4, seed5, seed6, num_in
     return all_images[0], all_images[1], all_images[2], all_images[3], all_images[4], all_images[5], "\n\n---\n\n".join(info_texts)
 
 # Create Gradio interface
-with gr.Blocks(title="Stable Diffusion Style Explorer", theme=gr.themes.Soft()) as demo:
+with gr.Blocks(title="Stable Diffusion Style Explorer") as demo:
+    
     gr.Markdown("""
     # 🎨 Stable Diffusion Style Explorer
     
@@ -540,13 +541,7 @@ with gr.Blocks(title="Stable Diffusion Style Explorer", theme=gr.themes.Soft())
 
 
 
-# Launch the app
 if __name__ == "__main__":
-    import gradio.routes
-
-    def disable_api_info(*args, **kwargs):
-        return {}
+    demo.launch(server_name="0.0.0.0", server_port=7860,
+    theme=gr.themes.Soft())
 
-    gradio.routes.get_api_info = disable_api_info
-    
-    demo.launch(server_name="0.0.0.0", server_port=7860)