lxl2023 commited on
Commit
016b491
1 Parent(s): 4c31b77

Add application file

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -9,8 +9,8 @@ import os
9
 
10
  OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
11
  OPENAI_API_BASE = os.getenv("OPENAI_API_BASE")
12
- # llm = ChatOpenAI(openai_api_key=OPENAI_API_KEY, temperature=0, model_name='gpt-3.5-turbo',openai_api_base=OPENAI_API_BASE)
13
- llm = AzureChatOpenAI(deployment_name="bitservice_chat_35", openai_api_base=OPENAI_API_BASE,
14
  openai_api_key=OPENAI_API_KEY, openai_api_version="2023-03-15-preview",
15
  model_name="gpt-3.5-turbo")
16
 
@@ -110,7 +110,7 @@ def segment(image, text):
110
  demo = gr.Interface(
111
  fn=segment,
112
  inputs=[
113
- gr.Image(type="filepath", shape=(200, 200), label="请选择一张图片"),
114
  gr.components.Textbox(label="请输入问题"),
115
  ],
116
  outputs=[gr.components.Textbox(label="答案", lines=4)])
 
9
 
10
  OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
11
  OPENAI_API_BASE = os.getenv("OPENAI_API_BASE")
12
+ DEP_NAME=os.getenv("deployment_name")
13
+ llm = AzureChatOpenAI(deployment_name=DEP_NAME, openai_api_base=OPENAI_API_BASE,
14
  openai_api_key=OPENAI_API_KEY, openai_api_version="2023-03-15-preview",
15
  model_name="gpt-3.5-turbo")
16
 
 
110
  demo = gr.Interface(
111
  fn=segment,
112
  inputs=[
113
+ gr.Image(type="filepath", label="请上传一张图片"),
114
  gr.components.Textbox(label="请输入问题"),
115
  ],
116
  outputs=[gr.components.Textbox(label="答案", lines=4)])