haohsiang commited on
Commit
051d335
·
1 Parent(s): c7b7c6b

change lora and add oauth button

Browse files
Files changed (1) hide show
  1. app.py +6 -5
app.py CHANGED
@@ -29,7 +29,7 @@ pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.conf
29
  # lora_path = "./tbh368-sdxl.safetensors"
30
  # pipe.load_lora_weights(lora_path, adapter_name="milton-glaser")
31
  # pipe.set_adapters(["milton-glaser"], adapter_weights=[1.0])
32
- pipe.load_lora_weights("KappaNeuro/konstantin-korovin-style", weight_name="Konstantin Korovin Style.safetensors", adapter_name="impressionism")
33
 
34
  # 載入 CVAW Corpus 資料
35
  cvaw_data = pd.read_csv('./CVAW_all_SD.csv', delimiter='\t')
@@ -125,7 +125,7 @@ def generate_images(history, conversation_times, last_genimg_times, generated_im
125
  last_genimg_times = conversation_times
126
 
127
  for i in range(4):
128
- img = pipe("Konstantin Korovin Style - , "+prompt,
129
  negative_prompt=neg_prompt,
130
  height=720, width=512,
131
  generator=generator[i],
@@ -278,7 +278,7 @@ css = """
278
  }
279
  """
280
 
281
- with gr.Blocks(theme=theme, css=css) as demo:
282
  generated_images = gr.State(value=None)
283
  meditation_flag = gr.State(value=True)
284
  last_genimg_times = gr.State(value=0)
@@ -286,6 +286,8 @@ with gr.Blocks(theme=theme, css=css) as demo:
286
  valence_scores = gr.State(value=[])
287
  arousal_scores = gr.State(value=[])
288
 
 
 
289
  loading_message = gr.Textbox(visible=False, show_label=False)
290
  with gr.Column(visible=False) as audio_interface:
291
  audio_player = gr.Audio(label="放鬆引導指導語", show_download_button=False, show_share_button=False, interactive=False)
@@ -309,8 +311,7 @@ with gr.Blocks(theme=theme, css=css) as demo:
309
  selected_image = gr.Image(interactive=False, show_share_button=False, label="你選擇的圖像")
310
  image_chat_input = gr.Textbox(label="這張圖像讓你產生了什麼樣的聯想?")
311
  image_chat_button = gr.Button("與聊天機器人分享", variant="primary")
312
- gr.LoginButton()
313
-
314
  # chatbot events handle
315
  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])
316
  msg.submit(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])
 
29
  # lora_path = "./tbh368-sdxl.safetensors"
30
  # pipe.load_lora_weights(lora_path, adapter_name="milton-glaser")
31
  # pipe.set_adapters(["milton-glaser"], adapter_weights=[1.0])
32
+ pipe.load_lora_weights("e-n-v-y/envyimpressionismxl01", weight_name="EnvyImpressionismXL01.safetensors", adapter_name="impressionism")
33
 
34
  # 載入 CVAW Corpus 資料
35
  cvaw_data = pd.read_csv('./CVAW_all_SD.csv', delimiter='\t')
 
125
  last_genimg_times = conversation_times
126
 
127
  for i in range(4):
128
+ img = pipe("modern digital impressionism, hard to understood, abstract, "+prompt,
129
  negative_prompt=neg_prompt,
130
  height=720, width=512,
131
  generator=generator[i],
 
278
  }
279
  """
280
 
281
+ with gr.Blocks(theme=theme, css=css, title='療癒對話機器人') as demo:
282
  generated_images = gr.State(value=None)
283
  meditation_flag = gr.State(value=True)
284
  last_genimg_times = gr.State(value=0)
 
286
  valence_scores = gr.State(value=[])
287
  arousal_scores = gr.State(value=[])
288
 
289
+ gr.Markdown("# 療癒對話機器人")
290
+ Login = gr.LoginButton(value="登入 HuggingFace", logout_value="登出", variant=primary, size=sm)
291
  loading_message = gr.Textbox(visible=False, show_label=False)
292
  with gr.Column(visible=False) as audio_interface:
293
  audio_player = gr.Audio(label="放鬆引導指導語", show_download_button=False, show_share_button=False, interactive=False)
 
311
  selected_image = gr.Image(interactive=False, show_share_button=False, label="你選擇的圖像")
312
  image_chat_input = gr.Textbox(label="這張圖像讓你產生了什麼樣的聯想?")
313
  image_chat_button = gr.Button("與聊天機器人分享", variant="primary")
314
+
 
315
  # chatbot events handle
316
  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])
317
  msg.submit(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])