Hugging Face's logo

Spaces:
shubham5027
/
Text_To_Image
Runtime error

App Files Community
shubham5027 commited on
Commit
5d68ba9
·
1 Parent(s): ce28d41

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -4,7 +4,7 @@ import torch
4
  import transformers
5
  from transformers import pipeline
6
  pipe = StableDiffusionPipeline.from_pretrained('runwayml/stable-diffusion-v1-5', torch_dtype=torch.float16)
7
- pipe = pipe.to("cuda")
8
 
9
  def generate_image(prompt):
10
 
 
4
  import transformers
5
  from transformers import pipeline
6
  pipe = StableDiffusionPipeline.from_pretrained('runwayml/stable-diffusion-v1-5', torch_dtype=torch.float16)
7
+
8
 
9
  def generate_image(prompt):
10