miittnnss
/

testmodel

Model card Files Files and versions

miittnnss commited on Sep 20, 2023

Commit

8b7694f

·

1 Parent(s): 95ae784

Update pipeline.py

Files changed (1) hide show

pipeline.py +1 -1

pipeline.py CHANGED Viewed

@@ -36,7 +36,7 @@ class PreTrainedPipeline():
         """
         Initialize model
         """
-        self.model = model = Generator.from_pretrained("miittnnss/testmodel")
     def __call__(self, inputs: str):
         """

         """
         Initialize model
         """
+        self.main = model = Generator.from_pretrained("miittnnss/testmodel")
     def __call__(self, inputs: str):
         """