shideqin commited on
Commit
a10c2cf
1 Parent(s): 21ed484

Update models/gpt_model.py

Browse files
Files changed (1) hide show
  1. models/gpt_model.py +1 -1
models/gpt_model.py CHANGED
@@ -8,7 +8,6 @@ if openai_api:
8
  class ImageToText:
9
  def __init__(self, gpt_version="gpt-3.5-turbo"):
10
  self.template = self.initialize_template()
11
- openai.api_key = os.getenv("OPENAI_API_KEY", "")
12
  self.gpt_version = gpt_version
13
 
14
  def initialize_template(self):
@@ -33,6 +32,7 @@ class ImageToText:
33
  print('\033[1;35m' + '*' * 100 + '\033[0m')
34
  print('\nStep4, Paragraph Summary with GPT-3:')
35
  print('\033[1;34m' + "Question:".ljust(10) + '\033[1;36m' + question + '\033[0m')
 
36
  completion = openai.ChatCompletion.create(
37
  model=self.gpt_version,
38
  messages = [
 
8
  class ImageToText:
9
  def __init__(self, gpt_version="gpt-3.5-turbo"):
10
  self.template = self.initialize_template()
 
11
  self.gpt_version = gpt_version
12
 
13
  def initialize_template(self):
 
32
  print('\033[1;35m' + '*' * 100 + '\033[0m')
33
  print('\nStep4, Paragraph Summary with GPT-3:')
34
  print('\033[1;34m' + "Question:".ljust(10) + '\033[1;36m' + question + '\033[0m')
35
+ openai.api_key = os.getenv("OPENAI_API_KEY", "")
36
  completion = openai.ChatCompletion.create(
37
  model=self.gpt_version,
38
  messages = [