Update app.py

app.py

@@ -64,6 +64,7 @@ from share_btn import community_icon_html, loading_icon_html, share_js
 HF_TOKEN_SD = os.environ.get('HF_TOKEN_SD')
 
 device = "cuda" if torch.cuda.is_available() else "cpu"
+print(f'device = {device}')
 
 def get_image_ext(img_bytes):
     w = imghdr.what("", img_bytes)