Update app.py

app.py

@@ -29,13 +29,7 @@ def get_types(cls_set: List[Type], component: str):
     return docset, types
 routes.get_types = get_types
 
-def cut(prom, out):
-    output = out[len(prom)-2:]
-    outout = re.sub("하하", "ㅎㅎ", output)
-    output = output.split('<|endoftext|>')[0]
-    output = output.split('\n')[0]
-    output = re.sub('[=+#/\:@*\"※ㆍ』‘|\\\<\>\(\)\[\]`\'…》\”\“\’·]', ' ', output)
-    return output
+
 # App code
 def mbti(x):
     t = loop.run_until_complete(translator.translate(x, src='ko', dest='en'))
@@ -59,9 +53,12 @@ def chat(x):
 		1.2,	# float, representing input in 'repetition_penalty' Slider component
 		fn_index=0
     )
-    result = cut(x, result)
-    
-    return result
+    output = result[len(x)-2:]
+    outout = re.sub("하하", "ㅎㅎ", output)
+    output = output.split('<|endoftext|>')[0]
+    output = output.split('\n')[0]
+    output = re.sub('[=+#/\:@*\"※ㆍ』‘|\\\<\>\(\)\[\]`\'…》\”\“\’·]', ' ', output)
+    return output
 
 def yn(x):
     result = gradio_client.predict(