1 files changed, 3 insertions, 5 deletions

diff --git a/testing/poe_test.py b/testing/poe_test.py
index 8d527879..809804f2 100644
--- a/testing/poe_test.py
+++ b/testing/poe_test.py
@@ -1,13 +1,11 @@
 from time import sleep
 
-import quora
+from  openai_rev import quora
 
 token = quora.Account.create(proxy=None, logging=True)
 print('token', token)
 
 sleep(2)
 
-for response in quora.StreamingCompletion.create(model='gpt-3.5-turbo',
-                                                 prompt='hello world',
-                                                 token=token):
-    print(response.completion.choices[0].text, end="", flush=True)
+for response in quora.StreamingCompletion.create(model='ChatGPT', prompt='hello world', token=token):
+    print(response.text, flush=True)