Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -8,7 +8,7 @@ import tempfile
|
|
| 8 |
import os
|
| 9 |
|
| 10 |
# 请将你的 OpenAI API 密钥放在这里
|
| 11 |
-
openai.api_key = '
|
| 12 |
|
| 13 |
# 功能一:拍照功能
|
| 14 |
st.title("拍照并处理图片")
|
|
@@ -31,13 +31,18 @@ if picture:
|
|
| 31 |
# 生成文字的提示词
|
| 32 |
prompt = "请从左到右描述图片中的物体。"
|
| 33 |
|
| 34 |
-
|
| 35 |
-
|
| 36 |
-
|
| 37 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 38 |
)
|
| 39 |
|
| 40 |
-
generated_text = response[
|
| 41 |
st.write("生成的文字结果:")
|
| 42 |
st.write(generated_text)
|
| 43 |
|
|
|
|
| 8 |
import os
|
| 9 |
|
| 10 |
# 请将你的 OpenAI API 密钥放在这里
|
| 11 |
+
openai.api_key = 'your-openai-api-key'
|
| 12 |
|
| 13 |
# 功能一:拍照功能
|
| 14 |
st.title("拍照并处理图片")
|
|
|
|
| 31 |
# 生成文字的提示词
|
| 32 |
prompt = "请从左到右描述图片中的物体。"
|
| 33 |
|
| 34 |
+
# 将图片的二进制数据转换为 base64 编码
|
| 35 |
+
import base64
|
| 36 |
+
img_base64 = base64.b64encode(img_byte_arr).decode('utf-8')
|
| 37 |
+
|
| 38 |
+
# 调用 OpenAI API 生成文字描述
|
| 39 |
+
response = openai.Completion.create(
|
| 40 |
+
engine="text-davinci-003",
|
| 41 |
+
prompt=f"{prompt}\n\n图片数据(base64编码):{img_base64}",
|
| 42 |
+
max_tokens=100
|
| 43 |
)
|
| 44 |
|
| 45 |
+
generated_text = response.choices[0].text.strip()
|
| 46 |
st.write("生成的文字结果:")
|
| 47 |
st.write(generated_text)
|
| 48 |
|