Upload MyTestPipeline

README.md

@@ -1,6 +1,6 @@
 ---
-library_name: transformers
 license: mit
+library_name: transformers
 ---
 
 # Model Card for Model ID

new_task.py

@@ -4,6 +4,12 @@ import tensorflow as tf
 import numpy as np
 
 class MyTestPipeline(TextGenerationPipeline):
+    def __init__(self, *args, **kwargs):
+      kwargs['trust_remote_code'] = True
+      self.trust_remote_code = True
+      super().__init__(*args, **kwargs)
+      self.trust_remote_code = True
+
     def preprocess(self, text, **kwargs):
         prompt = 'Answer the following question/statement without any explanation, do not abbreviate names, and give just the answer.'
         txt = f"<|user|>\n{prompt} {text}\n<|end|>\n<|assistant|>"