Update app.py

app.py

@@ -24,3 +24,14 @@ interface = gr.Interface(
 # Launch the interface
 if __name__ == "__main__":
     interface.launch()
+model = StableDiffusionPipeline.from_pretrained("SG161222/Realistic_Vision_V4.0")
+
+import torch
+from diffusers import StableDiffusionPipeline
+
+model = StableDiffusionPipeline.from_pretrained("SG161222/Realistic_Vision_V4.0").to("cuda")
+model.unet = torch.compile(model.unet)  # Speeds up inference
+
+def generate_image(prompt):
+    image = model(prompt).images[0]
+    return image