Update app.py

app.py

@@ -1,9 +1,13 @@
+import torch
 import streamlit as st
 from diffusers import StableDiffusion3Pipeline
 
 # Load the Diffusion pipeline
 pipeline = StableDiffusion3Pipeline.from_pretrained("stabilityai/stable-diffusion-3-medium-diffusers", torch_dtype=torch.float16)
 
+# Your other functions and main code here...
+
+
 def generate_prompt(prompt_text):
     # Generate response using the Diffusion model
     response = pipeline(prompt_text, top_p=0.9, max_length=100)[0]['generated_text']