kjqhdg

app.py

@@ -1,3 +1,4 @@
++import spaces
 import gradio as gr
 import torch
 from diffusers import DiffusionPipeline
@@ -8,6 +9,7 @@ pipe = DiffusionPipeline.from_pretrained(model_id)
 pipe.to("cuda")
 
 # Function to generate an image from text using diffusion
+[email protected]
 def generate_image(prompt):
     
     images = pipe(prompt=prompt, negative_prompt=neg_prompt,)