Update app.py

app.py

@@ -13,7 +13,7 @@ from datetime import datetime
 import random
 import torch
 import gradio as gr
-from diffusers import FluxPipeline, FlowMatchEulerDiscreteScheduler
+from diffusers import AutoPipelineForText2Image
 from PIL import Image
 import traceback
 import numpy as np
@@ -33,7 +33,7 @@ STYLE_PRESETS = {
 }
 
 # 固定模型配置
-FIXED_MODEL = "black-forest-labs/FLUX.1-dev"
+FIXED_MODEL = "aoxo/flux.1dev-abliterated"
 
 # 质量增强提示词（适配 NSFW）
 QUALITY_ENHANCERS = [
@@ -84,7 +84,7 @@ def initialize_model():
         print(f"📦 Loading fixed model: {FIXED_MODEL}")
         
         # ZeroGPU优化的模型加载
-        pipeline = FluxPipeline.from_pretrained(
+        pipeline = AutoPipelineForText2Image.from_pretrained(
             FIXED_MODEL,
             torch_dtype=torch.bfloat16 if torch.cuda.is_available() else torch.float32,
             # 添加内存优化参数