Spaces:

hshetty
/

movie-poster-generator

Runtime error

hshetty commited on Oct 18, 2022

Commit

25b84dd

1 Parent(s): c560c8f

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -24,7 +24,7 @@ def improve_image(img):
   return resp_img
 pipe = StableDiffusionPipeline.from_pretrained("CompVis/stable-diffusion-v1-4",use_auth_token= auth_token)
-pipe = pipe.to("cuda")
 def generate(celebrity, setting):
   # ANSWER HERE

   return resp_img
 pipe = StableDiffusionPipeline.from_pretrained("CompVis/stable-diffusion-v1-4",use_auth_token= auth_token)
+pipe = pipe.to(device)
 def generate(celebrity, setting):
   # ANSWER HERE