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):