awinml commited on
Commit
1602adf
1 Parent(s): e133330

Upload 2 files

Browse files
Files changed (2) hide show
  1. app.py +1 -1
  2. utils.py +1 -1
app.py CHANGED
@@ -121,7 +121,7 @@ if decoder_model == "GPT3 - (text-davinci-003)":
121
  api_key = save_key(openai_key)
122
  openai.api_key = api_key
123
  generated_text = gpt3(query_text, context_list)
124
- st.write(gpt3(generated_text))
125
 
126
  elif decoder_model == "T5":
127
  t5_pipeline = get_t5_model()
 
121
  api_key = save_key(openai_key)
122
  openai.api_key = api_key
123
  generated_text = gpt3(query_text, context_list)
124
+ st.write(generated_text)
125
 
126
  elif decoder_model == "T5":
127
  t5_pipeline = get_t5_model()
utils.py CHANGED
@@ -11,7 +11,7 @@ from transformers import (
11
  AutoModelForCausalLM,
12
  AutoModelForSeq2SeqLM,
13
  )
14
-
15
  import streamlit_scrollable_textbox as stx
16
 
17
 
 
11
  AutoModelForCausalLM,
12
  AutoModelForSeq2SeqLM,
13
  )
14
+ import openai
15
  import streamlit_scrollable_textbox as stx
16
 
17