change

SynonymEditor.py

@@ -13,6 +13,7 @@ class SynonymEditor:
         self.max_tokens = max_tokens
         self.temperature = temperature
         self.language = language
+        self.quote = '__NIE__' if (language == 'de') else '__QUOTE__'
 
     # Play with the prompts here and change the return index to change and see the effect of the prompt on the output quality
     # Note that the longer the prompt, higher the token used and hence the billing
@@ -55,7 +56,7 @@ class SynonymEditor:
     # Split the paragraph to preserve quotation marks
     def _split_into_sentences(self, text, few_shots):
         if (few_shots == False):
-            text = text.replace('"', '__QUOTE__')
+            text = text.replace('"', self.quote)
         text = re.sub(r'\s+', ' ', text)
         text = text.strip()
         sentences = sent_tokenize(text)
@@ -64,7 +65,7 @@ class SynonymEditor:
     def _post_process_sentence(self, text):
         print(text)
         print("==============")
-        return text.replace('__QUOTE__', '"')
+        return text.replace(self.quote, '"')
 
     # Preprocess the text, perform edit task and join back to get the original format
     def _edit_text(self, text, few_shots=False):