change lora, add OAuth button

app.py

@@ -22,13 +22,14 @@ client = OpenAI(api_key=api_key)
 # text to image 設定
 access_token = os.getenv('HF_TOKEN')
 model_id = "stabilityai/stable-diffusion-xl-base-1.0"
-pipe_t2i = StableDiffusionXLPipeline.from_pretrained(model_id, torch_dtype=torch.float16, use_safetensors=True, token=access_token).to("cuda")
-pipe_t2i.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe_t2i.scheduler.config)
+pipe = StableDiffusionXLPipeline.from_pretrained(model_id, torch_dtype=torch.float16, use_safetensors=True, token=access_token).to("cuda")
+pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)
 
-# 載入 LoRA 權重
-lora_path = "./tbh368-sdxl.safetensors"
-pipe_t2i.load_lora_weights(lora_path, adapter_name="milton-glaser")
-pipe_t2i.set_adapters(["milton-glaser"], adapter_weights=[1.0])
+# 載入 LoRA
+# lora_path = "./tbh368-sdxl.safetensors"
+# pipe.load_lora_weights(lora_path, adapter_name="milton-glaser")
+# pipe.set_adapters(["milton-glaser"], adapter_weights=[1.0])
+pipe.load_lora_weights("KappaNeuro/konstantin-korovin-style", weight_name="Konstantin Korovin Style.safetensors", adapter_name="impressionism")
 
 # 載入 CVAW Corpus 資料
 cvaw_data = pd.read_csv('./CVAW_all_SD.csv', delimiter='\t')
@@ -124,7 +125,7 @@ def generate_images(history, conversation_times, last_genimg_times, generated_im
   last_genimg_times = conversation_times
 
   for i in range(4):
-    img = pipe_t2i("style of Milton Glaser, healing image, "+prompt,
+    img = pipe("Konstantin Korovin Style - , "+prompt,
                 negative_prompt=neg_prompt,
                 height=720, width=512,
                 generator=generator[i],
@@ -308,6 +309,7 @@ with gr.Blocks(theme=theme, css=css) as demo:
       selected_image = gr.Image(interactive=False, show_share_button=False, label="你選擇的圖像")
       image_chat_input = gr.Textbox(label="這張圖像讓你產生了什麼樣的聯想？")
       image_chat_button = gr.Button("與聊天機器人分享", variant="primary")
+    gr.LoginButton()
 
   # chatbot events handle
   submit.click(handle_chat, [msg, chatbot, conversation_times, valence_scores, arousal_scores, meditation_flag], [chatbot, msg, submit, jump_to_med, meditation_buttons, conversation_times, valence_scores, arousal_scores, meditation_flag])