Update app.py

app.py

@@ -44,8 +44,9 @@ def mbti(x):
     
 def chat(x):
     x = f"{x}"
+    x = x.replace('friend','Human').replace('you','Assistant')
     result = gradio_client.predict(
-        x.replace('friend','Human').replace('you','Assistant'),
+        x,
         # str representing input in 'User input' Textbox component
 		0.92,	# float, representing input in 'Top-p (nucleus sampling)' Slider component
 		40,	    # int, representing input in 'Top-k (nucleus sampling)' Slider component
@@ -59,7 +60,6 @@ def chat(x):
     output = re.sub('하하','ㅎㅎ', output)
     output = output.split('띓')[0]
     output = output.split('<|endoftext|>')[0]
-    output = output.split('\n')[0]
     output = re.sub('[=+#/\:@*\"※ㆍ』‘|\\\<\>\(\)\[\]`\'…》\”\“\’·]', ' ', output)
     output = re.sub('[a-zA-Z]',' ',output)