brurei commited on
Commit
68f8297
·
verified ·
1 Parent(s): d0dd8bb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -2
app.py CHANGED
@@ -6,11 +6,14 @@ import random
6
  import time
7
  import sys
8
  import os
 
 
 
9
  from transformers import pipeline
10
  p = pipeline("automatic-speech-recognition")
11
 
12
 
13
- os.environ["OPENAI_API_KEY"] = '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'
14
 
15
  css = """
16
  .gradio-container {
@@ -184,7 +187,7 @@ def construct_index(directory_path):
184
 
185
  prompt_helper = PromptHelper(max_input_size, num_outputs, max_chunk_overlap, chunk_size_limit=chunk_size_limit)
186
 
187
- llm_predictor = LLMPredictor(llm=OpenAI(temperature=0.0, model_name="text-embedding-ada-002", max_tokens=num_outputs))
188
 
189
  documents = SimpleDirectoryReader(directory_path).load_data()
190
 
 
6
  import time
7
  import sys
8
  import os
9
+ import openai from dotenv import load_dotenv, find_dotenv
10
+
11
+
12
  from transformers import pipeline
13
  p = pipeline("automatic-speech-recognition")
14
 
15
 
16
+ openai.api_key = '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'
17
 
18
  css = """
19
  .gradio-container {
 
187
 
188
  prompt_helper = PromptHelper(max_input_size, num_outputs, max_chunk_overlap, chunk_size_limit=chunk_size_limit)
189
 
190
+ llm_predictor = LLMPredictor(llm=OpenAI(temperature=0.0, model_name="gpt-3.5-turbo", max_tokens=num_outputs))
191
 
192
  documents = SimpleDirectoryReader(directory_path).load_data()
193