Spaces:

polygraf-ai
/

article_writer

Runtime error

minko186 commited on Jul 18, 2024

Commit

ebc2abf

verified ·

1 Parent(s): d62b586

Update ai_generate.py

Files changed (1) hide show

ai_generate.py CHANGED Viewed

@@ -67,7 +67,7 @@ def generate(text, model, api):
             max_tokens=800,
             frequency_penalty=0.0
         )
-        return response[0].message.content
     elif model == "OpenAI GPT 4o":
         client = OpenAI(
             api_key=api,
@@ -80,4 +80,4 @@ def generate(text, model, api):
             max_tokens=800,
             frequency_penalty=0.0
         )
-        return response[0].message.content

             max_tokens=800,
             frequency_penalty=0.0
         )
+        return response.choices[0].message
     elif model == "OpenAI GPT 4o":
         client = OpenAI(
             api_key=api,
             max_tokens=800,
             frequency_penalty=0.0
         )
+        return response.choices[0].message