Yuchan5386 commited on
Commit
9c260f1
·
verified ·
1 Parent(s): 5d81911

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +16 -7
app.py CHANGED
@@ -182,10 +182,19 @@ def generate_text_better_sampling(model, prompt, max_len=100, max_gen=98, top_k=
182
 
183
  yield decoded_text
184
 
185
- def respond(user_input, chat_history): # chat_history는 무시해도 됨
186
- response = ""
187
- for partial in generate_text_better_sampling(model, user_input):
188
- response = partial
189
- yield response # 문자열만 넘기자!
190
- with gr.ChatInterface(fn=respond) as demo:
191
- demo.launch()
 
 
 
 
 
 
 
 
 
 
182
 
183
  yield decoded_text
184
 
185
+
186
+ nickname = "사용자"
187
+
188
+ def respond(message, chat_history):
189
+ message = message.replace("@사용자1@", nickname)
190
+ response = ""
191
+ for partial in generate_text_better_sampling(model, message):
192
+ response = partial
193
+ yield response
194
+
195
+ chat = gr.ChatInterface(
196
+ fn=respond,
197
+ title="InteractGPT",
198
+ )
199
+
200
+ chat.launch()