Update tasks/ref_in_gpt3.py

tasks/ref_in_gpt3.py

@@ -39,19 +39,34 @@ prompts.append("change the red apple to a green one, task: (referring editing),
 prompts.append("replace the dog with a cat, task: (referring editing), source: [dot], target: <cat>;")
 prompts.append("replace the red apple with a green one, task: (referring editing), source: [red apple], target: <green apple>;")
 
+openai.api_type = "azure"
+openai.api_base = "https://xdecoder.openai.azure.com/"
+openai.api_version = "2022-12-01"
 openai.api_key = os.environ["OPENAI_API_KEY"]
 
 def get_gpt3_response(prompt):
+    # response = openai.Completion.create(
+    #     model="text-davinci-003",
+    #     prompt=prompt,
+    #     temperature=0.7,
+    #     max_tokens=128,
+    #     top_p=1,
+    #     frequency_penalty=0,
+    #     presence_penalty=0
+    # )
+    
     response = openai.Completion.create(
-    model="text-davinci-003",
-    prompt=prompt,
-    temperature=0.7,
-    max_tokens=128,
-    top_p=1,
-    frequency_penalty=0,
-    presence_penalty=0
+      engine="text001",
+      prompt=prompt,
+      temperature=1,
+      max_tokens=100,
+      top_p=0.5,
+      frequency_penalty=0,
+      presence_penalty=0,
+      best_of=1,
+      stop=None
     )
-
+    
     return response
 
 def referring_inpainting_gpt3(model, image, instruction, *args, **kwargs):