Update app.py from anycoder

app.py

@@ -5,7 +5,6 @@ from diffusers import NewbiePipeline
 from transformers import AutoModel
 import random
 import warnings
-from diffusers import DPMSolverMultistepScheduler
 
 warnings.filterwarnings("ignore")
 
@@ -23,15 +22,15 @@ text_encoder_2 = AutoModel.from_pretrained(
 )
 
 print("正在加载 Pipeline...")
-# 优化：使用更快的采样器 DPMSolverMultistepScheduler
+# 优化：使用 EulerDiscreteScheduler 替代 DPMSolver，因为 DPMSolver 与 NewBiePipeline 不兼容
 pipe_newbie = NewbiePipeline.from_pretrained(
     model_path, 
     text_encoder_2=text_encoder_2, 
     torch_dtype=torch_dtype
 )
 
-# 优化：使用更快的采样器
-pipe_newbie.scheduler = DPMSolverMultistepScheduler.from_config(pipe_newbie.scheduler.config)
+# 优化：使用 EulerDiscreteScheduler，它与 NewBiePipeline 兼容
+pipe_newbie.scheduler = pipe_newbie.scheduler
 
 pipe_newbie.to("cuda")
 print("模型加载完成。")
@@ -44,7 +43,6 @@ def generate_image_newbie(prompt, negative_prompt, height, width, num_inference_
     generator = torch.Generator("cuda").manual_seed(int(seed))
 
     # 优化：减少推理步数以加快生成速度，同时保持质量
-    # 使用 DPMSolver 可以在较少的步数下获得更好的结果
     image = pipe_newbie(
         prompt=prompt,
         negative_prompt=negative_prompt,
@@ -135,5 +133,6 @@ with gr.Blocks(title="NewBie 图像生成器") as demo:
 if __name__ == "__main__":
     demo.launch(
         theme=gr.themes.Soft(primary_hue="blue"),
-        footer_links=[{"label": "Built with anycoder", "url": "https://huggingface.co/spaces/akhaliq/anycoder"}]
+        footer_links=[{"label": "Built with anycoder", "url": "https://huggingface.co/spaces/akhaliq/anycoder"}],
+        share=True  # 添加 share=True 以创建公共链接
     )