Update app.py

app.py

@@ -1,13 +1,25 @@
 import gradio as gr
 import numpy as np
 import random
-from diffusers import DiffusionPipeline
+from diffusers import StableDiffusionXLPipeline, EulerAncestralDiscreteScheduler
 import torch
 import spaces
 
+from huggingface_hub import hf_hub_download
+from safetensors.torch import load_file
+
 device = "cuda" 
 
-pipe = DiffusionPipeline.from_pretrained("aipicasso/emi-2",torch_dtype=torch.bfloat16)
+token=os.environ["TOKEN"]
+
+scheduler = EulerAncestralDiscreteScheduler.from_pretrained(model_id,subfolder="scheduler",token=token)
+pipe = StableDiffusionXLPipeline.from_pretrained(model_id, scheduler=scheduler, torch_dtype=torch.bfloat16,token=token)
+
+negative_ti_file = hf_hub_download(repo_id="Aikimi/unaestheticXL_Negative_TI", filename="unaestheticXLv31.safetensors")
+state_dict = load_file(negative_ti_file)
+pipe.load_textual_inversion(state_dict["clip_g"], token="unaestheticXLv31", text_encoder=pipe.text_encoder_2, tokenizer=pipe.tokenizer_2)
+pipe.load_textual_inversion(state_dict["clip_l"], token="unaestheticXLv31", text_encoder=pipe.text_encoder, tokenizer=pipe.tokenizer)
+
 pipe = pipe.to(device)
 
 
@@ -17,7 +29,7 @@ MAX_IMAGE_SIZE = 2048
 @spaces.GPU
 def infer(seed, randomize_seed, width, height, guidance_scale, num_inference_steps):
     prompt="1girl"
-    negative_prompt="bad hands"
+    negative_prompt="unaestheticXLv31"
     
     if randomize_seed:
         seed = random.randint(0, MAX_SEED)
@@ -55,7 +67,7 @@ with gr.Blocks(css=css) as demo:
             run_button = gr.Button("Run", scale=0)
         
         result = gr.Image(label="Result", show_label=False)
-        generated_prompt = gr.TextArea(show_label=False)
+        generated_prompt = gr.TextArea(label="Generated prompt", show_label=False)
         with gr.Accordion("Advanced Settings", open=False):
             
             seed = gr.Slider(