Update app.py

app.py

@@ -12,9 +12,7 @@ from authtoken import auth_token
 
 # Download stable diffusion model from Hugging Face
 modelid = "CompVis/stable-diffusion-v1-4"
-device = "cuda"
 stable_diffusion_model = StableDiffusionPipeline.from_pretrained(modelid, revision="fp16", torch_dtype=torch.float16, use_auth_token=auth_token)
-stable_diffusion_model.to(device)
 
 # Create a Streamlit app
 st.set_page_config(
@@ -35,7 +33,7 @@ def generate_image():
     if prompt:
         st.write("Generating image...")
         try:
-            with autocast(device):
+            with autocast():
                 image = stable_diffusion_model(prompt, guidance_scale=8.5)["sample"][0]
 
             # Display the generated image on the user interface