Merge pull request microsoft#13 from Django-Jiang/main

Update inference.py
HelloBroBro · Nov 6, 2023 · 809eea1 · 809eea1
2 parents 23676ee + 57ec124
commit 809eea1
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/inference.py b/inference.py
@@ -304,12 +304,12 @@ def call_openai_api(self, model, prompt):
  result = response['choices'][0]['text']
  else:
  response = openai.ChatCompletion.create(
- model='gpt-4-0613',
- messages=[
- {"role": "user", "content": prompt},
- ]
- )
- result = response['choices'][0]['message']['content']
+  model='gpt-4-0613',
+  messages=[
+  {"role": "user", "content": prompt},
+  ]
+  )
+  result = response['choices'][0]['message']['content']
  return result
 
  def pred_by_generation(self, input_text, model):