Update app.py

app.py

@@ -1,3 +1,4 @@
+import os
 import gradio as gr
 
 import torch
@@ -12,7 +13,7 @@ model_id = "runwayml/stable-diffusion-v1-5"
 # model_id = "CompVis/stable-diffusion-v1-4"
 
 has_cuda = torch.cuda.is_available()
-device = "cpu"
+
 if has_cuda:
     pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16, revision="fp16", use_auth_token=READ_TOKEN)
     device = "cuda"
@@ -70,9 +71,12 @@ with gr.Blocks() as demo:
     bt_make_image = gr.Button("Generate and image (takes about 10-15 minutes on CPU)", visible=False)
     
     image = gr.Image(label='Illustration for your story')
-    inference_steps = gr.Slider(5, 35, value=15, step=1, label="Num inference steps (more steps makes a better image but takes more time)")
+    inference_steps = gr.Slider(5, 30, value=15, step=1, label="Num inference steps (more steps makes a better image but takes more time)")
     
     bt_make_text.click(fn=generate_story, inputs=prompt, outputs=[story, summary, bt_make_image])
     bt_make_image.click(fn=generate_image, inputs=[summary, inference_steps], outputs=image)
 
-demo.launch(share=True)
+if READ_TOKEN:
+    demo.launch()
+else:
+    demo.launch(share=True, debug=True)