Upload MyTestPipeline

new_task.py

@@ -5,7 +5,7 @@ import numpy as np
 
 class MyTestPipeline(TextGenerationPipeline):
     def preprocess(self, text, **kwargs):
-        prompt = 'Answer the following question/statement without any explanation, do not abbreviate names, and give just the answer.'
+        prompt = 'Answer the following question/statement without any explanation, do not abbreviate names.'
         txt = f"<|user|>\n{prompt} {text}\n<|end|>\n<|assistant|>"
         return self.tokenizer(txt, return_tensors=self.framework)
     
@@ -35,5 +35,8 @@ class MyTestPipeline(TextGenerationPipeline):
         log_probs = np.round(np.exp(transition_scores.cpu().numpy()), 3)[0]
         guess_prob = np.product(log_probs)
 
+        if guess_prob >= 0.8:
+          guess_prob = 1.0
+
         return {'guess': guess_text, 'confidence': guess_prob}