change message

sintel-dev · Oct 1, 2024 · 9ecd5fa · 9ecd5fa
1 parent 7e57127
commit 9ecd5fa
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 8 deletions.
diff --git a/sigllm/primitives/prompting/gpt.py b/sigllm/primitives/prompting/gpt.py
@@ -88,7 +88,7 @@ def detect(self, X, **kwargs):
 
         all_responses, all_probs = [], []
         for text in tqdm(X):
-            message = ' '.join(PROMPTS['user_message'], text[0], self.sep)
+            message = ' '.join(PROMPTS['user_message'], text, self.sep)
             response = openai.ChatCompletion.create(
                 model=self.name,
                 messages=[

diff --git a/sigllm/primitives/prompting/huggingface.py b/sigllm/primitives/prompting/huggingface.py
@@ -115,13 +115,10 @@ def detect(self, X, **kwargs):
         all_responses, all_generate_ids = [], []
 
         for text in tqdm(X):
-            text = text.flatten().tolist()
-            message = [
-                ' '.join(
-                    (PROMPTS['system_message'],
-                     PROMPTS['user_message'],
-                        x,
-                        '[RESPONSE]')) for x in text]
+            message = ' '.join(PROMPTS['system_message'],
+                                PROMPTS['user_message'],
+                                text,
+                                '[RESPONSE]')
 
             input_length = len(self.tokenizer.encode(message[0]))