vilarin commited on
Commit
80f6e70
·
verified ·
1 Parent(s): acd0b30

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -52,6 +52,7 @@ def read(filename):
52
  return data
53
 
54
  def process_text(text_input, unit):
 
55
  if text_input and endpoints == 'OPENAI':
56
  client = OpenAI(api_key=api_key)
57
  completion = client.chat.completions.create(
@@ -77,6 +78,7 @@ def encode_image_to_base64(image_input):
77
  return img_str
78
 
79
  def process_image(image_input, unit):
 
80
  if image_input is not None and endpoints == 'OPENAI':
81
  client = OpenAI(api_key=api_key)
82
  #with open(image_input.name, "rb") as f:
@@ -107,7 +109,6 @@ def process_image(image_input, unit):
107
  return response.text
108
 
109
  def main(api_key="", text_input="", image_input=None, unit=""):
110
- endpoints = endpoints(api_key)
111
  SYS_PROMPT = read('system_prompt.txt')
112
  if text_input and image_input is None:
113
  return process_text(text_input,unit)
 
52
  return data
53
 
54
  def process_text(text_input, unit):
55
+ endpoints = endpoints(api_key)
56
  if text_input and endpoints == 'OPENAI':
57
  client = OpenAI(api_key=api_key)
58
  completion = client.chat.completions.create(
 
78
  return img_str
79
 
80
  def process_image(image_input, unit):
81
+ endpoints = endpoints(api_key)
82
  if image_input is not None and endpoints == 'OPENAI':
83
  client = OpenAI(api_key=api_key)
84
  #with open(image_input.name, "rb") as f:
 
109
  return response.text
110
 
111
  def main(api_key="", text_input="", image_input=None, unit=""):
 
112
  SYS_PROMPT = read('system_prompt.txt')
113
  if text_input and image_input is None:
114
  return process_text(text_input,unit)