Update app.py

app.py

@@ -4,12 +4,13 @@ from PIL import Image
 import numpy as np
 import cv2
 from diffusers import StableDiffusionPipeline
+from huggingface_hub import login
 
 
 
 # Setup the model
 device = "cuda" if torch.cuda.is_available() else "cpu"
-model_id = "stabilityai/stable-diffusion-2-1"
+model_id = "s3nh/artwork-arcane-stable-diffusion"
 pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16 if device == "cuda" else torch.float32, use_auth_token=True)
 pipe = pipe.to(device)
 
@@ -106,4 +107,4 @@ with gr.Blocks(css=css) as demo:
         outputs=[result]
     )
 
-demo.queue().launch()
+demo.queue().launch()