Update app.py

app.py

@@ -4,12 +4,16 @@ import numpy as np
 import re
 import torch
 
-tokenizer = BertTokenizer.from_pretrained(r'morror_art_test/model')
-model = BertForSequenceClassification.from_pretrained(r'morror_art_test/model', num_labels = 6)
+tokenizer = BertTokenizer.from_pretrained(r'./model')
+model = BertForSequenceClassification.from_pretrained(r'./model', num_labels = 6)
 
 def preprocess(temp):
     temp = re.sub(u"\n\n", "\n", temp)
     temp = re.sub(u"(^\n)|(\n$)", "", temp)
+    temp = re.sub(u",", "，", temp)
+    temp = re.sub(u"?", "？", temp)
+    temp = re.sub(u"!", "！", temp)
+    temp = re.sub(u".", "。", temp)
     temp = re.sub('[^\u4e00-\u9fa5，。？！\n]+', '', temp)
     temp = re.sub(u"\n", "，", temp)
     for _ in range(int(len(temp) / 2)):
@@ -17,7 +21,6 @@ def preprocess(temp):
         temp = re.sub(u"，！|！，", "！", temp)
         temp = re.sub(u"，？|？，", "？", temp)
         temp = re.sub(u"，。|。，", "。", temp)
-    # time.sleep(1)
     temp = temp.strip('，')
 
     return temp