Update app.py

app.py

@@ -1,15 +1,15 @@
-from diffusers import AutoPipelineForText2Image
+from diffusers import StableDiffusionImg2ImgPipeline
 import torch
 import gradio as gr
 
 
 def get_model():
-    pipe = AutoPipelineForText2Image.from_pretrained("stabilityai/sdxl-turbo", variant="fp16")
+    pipe = StableDiffusionImg2ImgPipeline.from_pretrained("stabilityai/sdxl-turbo", variant="fp16")
+    pipe.to("cpu")
+    pipe.enable_model_cpu_offload()
     return pipe
 
 model = get_model()
-model.to("cpu")
-model.enable_model_cpu_offload()
 
 
 def return_image(text,inf_steps,gScale):