Update app.py

app.py

@@ -2,6 +2,8 @@ import pip
 pip.main(['install', 'torch'])
 pip.main(['install', 'transformers'])
 
+import re
+
 import torch
 import gradio as gr
 import transformers
@@ -28,7 +30,7 @@ def inference(prompt_inputs):
     prompt_inputs = prompt_inputs.replace('\n', ' ')
 
     # prompt 구두점 단위로 분리하기
-    prompt_list = prompt_inputs.split('.|!|?')
+    prompt_list = re.split('[.?!]+', prompt_inputs)
 
     class_id_list = []