Update ImageToTextPipelineTests.test_small_model_tf (huggingface#19785

) * update expected values for the correct TF checkpoint * Run test * Clean up * fix Co-authored-by: ydshieh <[email protected]>
chenhxbj · Oct 21, 2022 · 3aaabaa · 3aaabaa
1 parent 7487829
commit 3aaabaa
Showing 1 changed file with 5 additions and 20 deletions.
diff --git a/tests/pipelines/test_pipelines_image_to_text.py b/tests/pipelines/test_pipelines_image_to_text.py
@@ -64,12 +64,7 @@ def test_small_model_tf(self):
             outputs,
             [
                 {
-                    "generated_text": (
-                        " intermedi intermedi intermedi intermedi intermedi "
-                        "explorer explorer explorer explorer explorer explorer "
-                        "explorer medicine medicine medicine medicine medicine "
-                        "medicine medicine"
-                    )
+                    "generated_text": "growthgrowthgrowthgrowthgrowthgrowthgrowthgrowthgrowthgrowthgrowthgrowthgrowthgrowthgrowthgrowthgrowthGOGO"
                 },
             ],
         )
@@ -80,23 +75,13 @@ def test_small_model_tf(self):
             [
                 [
                     {
-                        "generated_text": (
-                            " intermedi intermedi intermedi intermedi intermedi "
-                            "explorer explorer explorer explorer explorer explorer "
-                            "explorer medicine medicine medicine medicine medicine "
-                            "medicine medicine"
-                        )
-                    },
+                        "generated_text": "growthgrowthgrowthgrowthgrowthgrowthgrowthgrowthgrowthgrowthgrowthgrowthgrowthgrowthgrowthgrowthgrowthGOGO"
+                    }
                 ],
                 [
                     {
-                        "generated_text": (
-                            " intermedi intermedi intermedi intermedi intermedi "
-                            "explorer explorer explorer explorer explorer explorer "
-                            "explorer medicine medicine medicine medicine medicine "
-                            "medicine medicine"
-                        )
-                    },
+                        "generated_text": "growthgrowthgrowthgrowthgrowthgrowthgrowthgrowthgrowthgrowthgrowthgrowthgrowthgrowthgrowthgrowthgrowthGOGO"
+                    }
                 ],
             ],
         )