change

SynonymEditor.py

@@ -13,7 +13,7 @@ class SynonymEditor:
         self.max_tokens = max_tokens
         self.temperature = temperature
         self.language = language
-        self.quote = '__NIE__' if (language == 'de') else '__QUOTE__'
+        self.quote = '__ZITIEREN__' 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
@@ -25,7 +25,7 @@ class SynonymEditor:
             else:
                 prompt = "Replace exactly one word with a synonym while preserving the overall sentence structure and meaning.\n" + \
                     few_shots + "\nInput:" + sentence + " Output:"
-        elif "__QUOTE__" in sentence:
+        elif self.quote in sentence:
             if (self.language == "de"):
                 prompt = 'Modernisiere den deutschen Text. Fasse direkte Reden NIE zusammen.\n'+sentence+'\n'
             else: