Update app.py

app.py

@@ -1,5 +1,20 @@
 import gradio as gr
 from transformers import pipeline
++import spaces
+from diffusers import DiffusionPipeline
+
+pipe = DiffusionPipeline.from_pretrained(...)
+pipe.to('cuda')
+
+[email protected]
+def generate(prompt):
+    return pipe(prompt).images
+
+gr.Interface(
+    fn=generate,
+    inputs=gr.Text(),
+    outputs=gr.Gallery(),
+).launch()
 
 # Initialiser le générateur de texte avec GPT-2
 generator = pipeline('text-generation', model='gpt2')