Hugging Face's logo

Spaces:
pratikshahp
/
text-to-image
Sleeping

App Files Community
pratikshahp commited on
Commit
b02f2f6
•
1 Parent(s): 4f953de

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -5,7 +5,7 @@ from diffusers import StableDiffusionPipeline
5
 
6
  def generate_image(input_text):
7
  # Load Diffusion pipeline
8
- pipe = StableDiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5", torch_dtype=torch.float16)
9
  # Generate image from text prompt
10
  prompt = input_text
11
  generated_image = pipe(prompt).images[0]
 
5
 
6
  def generate_image(input_text):
7
  # Load Diffusion pipeline
8
+ pipe = StableDiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5", torch_dtype=torch.float32)
9
  # Generate image from text prompt
10
  prompt = input_text
11
  generated_image = pipe(prompt).images[0]