selfhostyourtech/reflector
2
0
Fork 0
mirror of https://github.com/Monadical-SAS/reflector.git synced 2025-12-20 20:29:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #30 from Monadical-SAS/feat/gokul

Browse Source 
Remove key
This commit is contained in:
projects-g
2023-07-14 23:06:24 +05:30
committed by GitHub
parent 0bfb5db6a3 03da9f61ca
commit 0f3beb27a7
1 changed files with 9 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
trials/openai_endpoint.py
View File

@@ -3,7 +3,7 @@
import openai import openai
openai.api_key = "***REMOVED***" openai.api_key = ""
# to caption, user prompt used : "caption this conversation" # to caption, user prompt used : "caption this conversation"
# max_tokens=20 # max_tokens=20
@@ -21,10 +21,16 @@ conversation = [
"content": "summarize this conversation in a few sentences by taking key points"} "content": "summarize this conversation in a few sentences by taking key points"}
] ]
response = openai.ChatCompletion.create(model="gpt-3.5-turbo", model = "gpt-3.5-turbo"
response = openai.ChatCompletion.create(model=model,
messages=conversation, messages=conversation,
n=1, n=1,
max_tokens=300) max_tokens=300)
caption = response.choices[0].message.content.strip() # Try finetuned model
# model = "davinci:ft-personal-2023-07-14-10-43-51"
# response = openai.Completion.create(model=model,
# prompt=sample_chunks[0] + " -> ")
caption = response.choices[0]
print(caption) print(caption)