Update app.py

app.py

@@ -1,12 +1,10 @@
-from diffusers import StableDiffusionPipeline
+from diffusers import DiffusionPipeline
 import streamlit as st
 import torch
 
-model_id = "AMAN-B/Demo-Dreambooth"
-pipe = StableDiffusionPipeline.from_pretrained(model_id,  torch_dtype=torch.float16).to("cuda")
-
+generator = DiffusionPipeline.from_pretrained("AMAN-B/Demo-Dreambooth").to("cuda")
 
 prompt = st.text_area("enter prompt")
 if text:
-	image = pipe(prompt, num_inference_steps=50, height=512, width=512, guidance_scale=7.5).images[0]
+	image = generator(prompt, num_inference_steps=50, height=512, width=512, guidance_scale=7.5).images[0]
 	st.json(image)