ake178178 commited on
Commit
2fc8dce
·
verified ·
1 Parent(s): 7819b33

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +11 -6
app.py CHANGED
@@ -8,7 +8,7 @@ import tempfile
8
  import os
9
 
10
  # 请将你的 OpenAI API 密钥放在这里
11
- openai.api_key = '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'
12
 
13
  # 功能一:拍照功能
14
  st.title("拍照并处理图片")
@@ -31,13 +31,18 @@ if picture:
31
  # 生成文字的提示词
32
  prompt = "请从左到右描述图片中的物体。"
33
 
34
- response = openai.Image.create(
35
- image=img_byte_arr,
36
- size="300x300",
37
- prompt=prompt
 
 
 
 
 
38
  )
39
 
40
- generated_text = response['data']['text']
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