Import os for env vars

app.py

@@ -6,16 +6,14 @@ import gradio
 from diffusers import StableDiffusionPipeline as pipeline
 from accelerate import init_empty_weights
 import torch
+import os
 
-from PIL import Image
-import matplotlib.pyplot as plot
-
+api_key = os.environ['api_key']
 my_token = api_key
 
 with init_empty_weights():
   pipe = pipeline.from_pretrained("CompVis/stable-diffusion-v1-4", revision="fp16", torch_dtype=torch.float16, use_auth_token=my_token).to("cuda")
 
-
 def image_prompt(thing):
   prompt = "A children's drawing of a " + thing
   return pipe(prompt=prompt, height=512, width=664).images[0]