Create app.py

app.py

@@ -0,0 +1,28 @@
+import gradio as gr
+import torch
+from diffusers import AutoPipelineForText2Image
+
+# Install required libraries
+!pip install diffusers transformers accelerate --upgrade
+
+# Load pre-trained model and move it to GPU
+pipe = AutoPipelineForText2Image.from_pretrained("stabilityai/sdxl-turbo", torch_dtype=torch.float16, variant="fp16")
+pipe.to("cuda")
+
+# Define function to generate image based on input prompt
+def generate_image(prompt):
+    image = pipe(prompt=prompt, num_inference_steps=1, guidance_scale=0.0).images[0]
+    return image
+
+# Create Gradio interface
+input_text = gr.inputs.Textbox(lines=5, label="Enter the prompt")
+output_image = gr.outputs.Image(label="Generated Image")
+
+# Define Gradio app
+gr.Interface(
+    generate_image, 
+    inputs=input_text, 
+    outputs=output_image,
+    title="Stable Diffusion SDXL Turbo",
+    description="Enter a prompt and generate an image based on the input prompt"
+).launch()