From 57ec1240c9ba914554479e7c9cb8612233471910 Mon Sep 17 00:00:00 2001
From: "F. J" <43953876+Django-Jiang@users.noreply.github.com>
Date: Thu, 2 Nov 2023 13:46:50 -0700
Subject: [PATCH] Update inference.py

fix openai call
---
 inference.py | 12 ++++++------
 1 file changed, 6 insertions(+), 6 deletions(-)

diff --git a/inference.py b/inference.py
index ea8870f..10a728c 100644
--- 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):