siddh4rth commited on
Commit
f18d0df
·
1 Parent(s): 09d4b23

updated app file

Browse files
Files changed (1) hide show
  1. app.py +10 -18
app.py CHANGED
@@ -63,25 +63,17 @@ def retrieve(query):
63
  # then we complete the context-infused query
64
  def complete(prompt):
65
  # query text-davinci-003
66
- response = openai.ChatCompletion.create(
67
- model="gpt-4",
68
- messages=[
69
- {
70
- "role": "system",
71
- "content": "Answer the question based on the context below and the document provided."
72
- },
73
- {
74
- "role": "user",
75
- "content": prompt
76
- }
77
- ],
78
- temperature=0.5,
79
- max_tokens=128,
80
- top_p=1.0,
81
- frequency_penalty=0.0,
82
- presence_penalty=0.0
83
  )
84
- return response.choices[0].message.content.strip()
85
 
86
  def greet(query):
87
  # first we retrieve relevant items from Pinecone
 
63
  # then we complete the context-infused query
64
  def complete(prompt):
65
  # query text-davinci-003
66
+ res = openai.Completion.create(
67
+ engine='text-davinci-003',
68
+ prompt=prompt,
69
+ temperature=0,
70
+ max_tokens=500,
71
+ top_p=1,
72
+ frequency_penalty=0,
73
+ presence_penalty=0,
74
+ stop=None
 
 
 
 
 
 
 
 
75
  )
76
+ return res['choices'][0]['text'].strip()
77
 
78
  def greet(query):
79
  # first we retrieve relevant items from Pinecone