Update requirements.txt

requirements.txt

@@ -1,39 +1,5 @@
-import spaces
-import torch
-import gradio as gr
-from diffusers import DiffusionPipeline
-
-MODEL = "black-forest-labs/FLUX.1-schnell"
-
-pipe = DiffusionPipeline.from_pretrained(
-    MODEL,
-    torch_dtype=torch.bfloat16
-).to("cuda")
-
-
-@spaces.GPU(duration=60)
-def generate_image(prompt):
-    image = pipe(
-        prompt=prompt,
-        guidance_scale=0.0,
-        num_inference_steps=4,
-        height=768,
-        width=768,
-        max_sequence_length=256,
-    ).images[0]
-
-    return image
-
-
-demo = gr.Interface(
-    fn=generate_image,
-    inputs=gr.Textbox(
-        label="Image Prompt",
-        placeholder="Describe the image you want..."
-    ),
-    outputs=gr.Image(label="Generated Image"),
-    title="AI Image Generator",
-    description="Enter a prompt and generate an image."
-)
-
-demo.launch()
+diffusers
+transformers
+accelerate
+safetensors
+sentencepiece