fix: docs

baidubce · Jan 11, 2024 · 7bbe09e · 7bbe09e
1 parent 8111d1a
commit 7bbe09e
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 3 deletions.
diff --git a/docs/inference.md b/docs/inference.md
@@ -191,6 +191,16 @@ img_data = resp["body"]["data"][0]["image"]
 img = Image.open(io.BytesIO(img_data))
 ```
 
+#### **图生文**
+千帆平台也提供了图+文生文功能，千帆SDK支持用户调用SDK来获取结果，以快速集成多模态能力到大模型应用中。
+
+以下是一个使用示例
+```python
+i2t = qianfan.Image2Text(endpoint="....")
+resp = i2t.do(prompt="Rag doll cat", "9j/4AAQSkZJRgABAQAAAQABAAD/xxxxx")
+print(resp["result"])
+```
+
 #### **批量推理**
 
 上述模型均提供了 `batch_do` 和异步的 `abatch_do` 方法，方便用户批量进行推理，并通过 `worker_num` 来控制并发量。

diff --git a/src/qianfan/resources/images/image2text.py b/src/qianfan/resources/images/image2text.py
@@ -157,7 +157,7 @@ def do(
         as follows:
 
         ```
-        Text2Image().do(prompt = ..., steps=20)
+        Image2Text(endpoint="").do(prompt="", image="", xxx=vvv)
         ```
 
         """
@@ -220,7 +220,7 @@ async def ado(
         as follows:
 
         ```
-        Text2Image().do(prompt = ..., steps=20)
+        Image2Text(endpoint="").ado(prompt="", image="", xx=vv)
         ```
 
         """

diff --git a/src/qianfan/tests/image2text_test.py b/src/qianfan/tests/image2text_test.py
@@ -13,7 +13,7 @@
 # limitations under the License.
 
 """
-    Unit test for Txt2Image
+    Unit test for Image2Text
 """
 
 import time