Update app.py

app.py

@@ -10,10 +10,10 @@ from dataclasses import dataclass
 import math
 from typing import Callable
 import spaces
-from diffusers import DiffusionPipeline, AutoencoderTiny, AutoencoderKL
+from diffusers import ZImagePipeline, DiffusionPipeline, AutoencoderTiny, AutoencoderKL
 from diffusers.schedulers import FlowMatchEulerDiscreteScheduler
-from transformers import CLIPModel, CLIPProcessor, CLIPTextModel, CLIPTokenizer, CLIPConfig, T5EncoderModel, T5Tokenizer
-from diffusers.models.transformers import FluxTransformer2DModel
+#from transformers import CLIPModel, CLIPProcessor, CLIPTextModel, CLIPTokenizer, CLIPConfig, T5EncoderModel, T5Tokenizer
+#from diffusers.models.transformers import FluxTransformer2DModel
 import copy
 import random
 import time