Spaces:

hshetty
/

movie-poster-generator

Runtime error

hshetty commited on Oct 17, 2022

Commit

7842af6

1 Parent(s): ee23095

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -23,7 +23,7 @@ def improve_image(img):
   resp_img = gr.processing_utils.decode_base64_to_image((resp_obj.json())['data'][0])
   return resp_img
-pipe = StableDiffusionPipeline.from_pretrained("CompVis/stable-diffusion-v1-4")
 pipe = pipe.to("cuda")
 def generate(celebrity, setting):

   resp_img = gr.processing_utils.decode_base64_to_image((resp_obj.json())['data'][0])
   return resp_img
+pipe = StableDiffusionPipeline.from_pretrained("CompVis/stable-diffusion-v1-4",use_auth_token=True)
 pipe = pipe.to("cuda")
 def generate(celebrity, setting):