Update app.py

app.py

@@ -3,15 +3,14 @@ import numpy as np
 import random
 import spaces
 import torch
-from diffusers import  QwenImagePipeline
+from diffusers import  DiffusionPipeline
 
 dtype = torch.bfloat16
 device = "cuda" if torch.cuda.is_available() else "cpu"
 
 
-
 # Load the model pipeline
-pipe = QwenImagePipeline.from_pretrained("Qwen/Qwen-Image", torch_dtype=dtype).to(device)
+pipe = DiffusionPipeline.from_pretrained("Qwen/Qwen-Image", torch_dtype=dtype).to(device)
 
 torch.cuda.empty_cache()
 
@@ -30,8 +29,7 @@ def infer(prompt, seed=42, randomize_seed=False, width=1024, height=1024, guidan
     if lora_id and lora_id.strip() != "":
         pipe.unload_lora_weights()
         pipe.load_lora_weights(lora_id.strip())
-
-
+    
     try:
         image = pipe(
         prompt=prompt,