Upload main.py

main.py

@@ -200,7 +200,8 @@ class InferenceClass:
         
         if bert_class=="SLM":
             model_output = self.call_gpt(chat_history)
-            model_output = model_output.replace("<RAG data>","<keywords>")
+            model_output = model_output.replace("<RAG data>","<keywords>").replace('晚上晚上','晚上').replace("ShoushanLover'sObservatory","Shoushan Lover's Observatory").replace('SizihwanBay','Sizihwan Bay')
+
             print(model_output)
             tmp = {'from': 'gpt','Next':False}
             outs = model_output.split(';\n')