bramhampton commited on
Commit
9ce542b
1 Parent(s): 3c37c4f

Only get generated text from object

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -11,4 +11,4 @@ def query(payload):
11
  promptBox = st.text_input("Please provide a starting point for your prompt")
12
 
13
  if (len(promptBox) > 0):
14
- st.write(query(promptBox))
 
11
  promptBox = st.text_input("Please provide a starting point for your prompt")
12
 
13
  if (len(promptBox) > 0):
14
+ st.write(query(promptBox)[0]["generated_text"])