Spaces:

Anustup
/

Youtube-QNA

Sleeping

Anustup commited on Feb 14

Commit

3e85d93

•

1 Parent(s): aa5998e

Update utils.py

Files changed (1) hide show

utils.py CHANGED Viewed

@@ -24,6 +24,7 @@ def create_open_ai_query(input_query, system_message=None, model_engine=OPEN_AI_
     payload = {
         'model': model_engine,
         'messages': messages,
     }
     response = requests.post(openai_url, headers=headers, data=json.dumps(payload))
     if response.status_code == 200 and 'choices' in response.json():

     payload = {
         'model': model_engine,
         'messages': messages,
+        'stream': True
     }
     response = requests.post(openai_url, headers=headers, data=json.dumps(payload))
     if response.status_code == 200 and 'choices' in response.json():