robertselvam commited on
Commit
1bc0310
1 Parent(s): 83fd3e6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -105,7 +105,7 @@ class KeyValueExtractor:
105
 
106
  # Load the summarization chain using the ChatOpenAI language model
107
  chain = load_summarize_chain(
108
- llm = AzureChatOpenAI(azure_deployment = "ChatGPT"),
109
  chain_type="refine",
110
  question_prompt=prompt,
111
  refine_prompt=refine_prompt,
@@ -129,7 +129,7 @@ class KeyValueExtractor:
129
 
130
  # Call OpenAI GPT-3.5-turbo
131
  chat_completion = self.client.chat.completions.create(
132
- model = "ChatGPT",
133
  messages = conversation,
134
  max_tokens=1000,
135
  temperature=0
@@ -167,7 +167,7 @@ class KeyValueExtractor:
167
 
168
  # Call OpenAI GPT-3.5-turbo
169
  chat_completion = self.client.chat.completions.create(
170
- model = "ChatGPT",
171
  messages = conversation,
172
  max_tokens=1000,
173
  temperature=0
@@ -318,7 +318,7 @@ class KeyValueExtractor:
318
 
319
  # Load the summarization chain using the ChatOpenAI language model
320
  chain = load_summarize_chain(
321
- llm = AzureChatOpenAI(azure_deployment = "ChatGPT"),
322
  chain_type="refine",
323
  question_prompt=prompt,
324
  refine_prompt=refine_prompt,
 
105
 
106
  # Load the summarization chain using the ChatOpenAI language model
107
  chain = load_summarize_chain(
108
+ llm = AzureChatOpenAI(azure_deployment = "GPT-3"),
109
  chain_type="refine",
110
  question_prompt=prompt,
111
  refine_prompt=refine_prompt,
 
129
 
130
  # Call OpenAI GPT-3.5-turbo
131
  chat_completion = self.client.chat.completions.create(
132
+ model = "GPT-3",
133
  messages = conversation,
134
  max_tokens=1000,
135
  temperature=0
 
167
 
168
  # Call OpenAI GPT-3.5-turbo
169
  chat_completion = self.client.chat.completions.create(
170
+ model = "GPT-3",
171
  messages = conversation,
172
  max_tokens=1000,
173
  temperature=0
 
318
 
319
  # Load the summarization chain using the ChatOpenAI language model
320
  chain = load_summarize_chain(
321
+ llm = AzureChatOpenAI(azure_deployment = "GPT-3"),
322
  chain_type="refine",
323
  question_prompt=prompt,
324
  refine_prompt=refine_prompt,