Spaces:

Sabbah13
/

text_transcribation_diarization_and_summarization

Paused

App Files Files Community

Sabbah13 commited on Aug 15

Commit

41e2dda

•

1 Parent(s): a93e42f

Update openai_requests.py

Files changed (1) hide show

openai_requests.py +2 -2

openai_requests.py CHANGED Viewed

@@ -4,7 +4,7 @@ import requests
 import json
 import math
-def get_completion_from_openai(prompt, max_tokens = None):
     url = os.getenv('OPENAI_COMPLETION_URL')
     headers = {
@@ -14,7 +14,7 @@ def get_completion_from_openai(prompt, max_tokens = None):
     response = requests.post(url,
                                  json={
-                                     "model": os.getenv('OPENAI_MODEL'),
                                      "max_tokens": max_tokens,
                                      "messages": [
                                       {

 import json
 import math
+def get_completion_from_openai(prompt, max_tokens = None, model):
     url = os.getenv('OPENAI_COMPLETION_URL')
     headers = {
     response = requests.post(url,
                                  json={
+                                     "model": model,
                                      "max_tokens": max_tokens,
                                      "messages": [
                                       {