1 files changed, 4 insertions, 9 deletions

diff --git a/testing/quora_test_2.py b/testing/quora_test_2.py
index d5316946..5d06f9ed 100644
--- a/testing/quora_test_2.py
+++ b/testing/quora_test_2.py
@@ -1,17 +1,12 @@
-import quora
+from openai_rev import quora
 
 token = quora.Account.create(logging=True, enable_bot_creation=True)
 
 model = quora.Model.create(
-    token=token,
-    model='gpt-3.5-turbo',  # or claude-instant-v1.0
-    system_prompt='you are ChatGPT a large language model ...'
+    token=token, model='ChatGPT', system_prompt='you are ChatGPT a large language model ...'  # or claude-instant-v1.0
 )
 
 print(model.name)
 
-for response in quora.StreamingCompletion.create(
-        custom_model=model.name,
-        prompt='hello world',
-        token=token):
-    print(response.completion.choices[0].text)
+for response in quora.StreamingCompletion.create(custom_model=model.name, prompt='hello world', token=token):
+    print(response.text)