awacke1 commited on
Commit
1279ad7
·
1 Parent(s): 2158acb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -121,11 +121,12 @@ generator2 = gr.Interface.load("huggingface/EleutherAI/gpt-neo-2.7B", api_key=HF
121
  generator3 = gr.Interface.load("huggingface/EleutherAI/gpt-j-6B", api_key=HF_TOKEN)
122
 
123
  #MemoryChange=tbOutput.change(persist_memory,inputs=[tbOutput],outputs=gr.Textbox(label="PersistMemoryOutput"))
 
124
 
125
  parallelModel = gr.Parallel(generator1,
126
  generator2,
127
  generator3,
128
- inputs = gr.inputs.Textbox(lines=5, label="Enter a sentence to get another sentence."),
129
  examples=examples,
130
  title="Mindfulness Story Generation with Persistent Dataset Memory",
131
  description=f"Mindfulness Story Generation with Persistent Dataset Memory",
@@ -134,6 +135,6 @@ parallelModel = gr.Parallel(generator1,
134
 
135
  tbMemoryOutput = gr.Textbox(label="Memory Output")
136
  btnSave = gr.Button("Save")
137
- btnSave.click(fn=persist_memory, inputs=tbOutput, outputs=tbMemoryOutput)
138
 
139
  parallelModel.launch(share=False)
 
121
  generator3 = gr.Interface.load("huggingface/EleutherAI/gpt-j-6B", api_key=HF_TOKEN)
122
 
123
  #MemoryChange=tbOutput.change(persist_memory,inputs=[tbOutput],outputs=gr.Textbox(label="PersistMemoryOutput"))
124
+ SplitterInputBox = gr.inputs.Textbox(lines=5, label="Enter a sentence to get another sentence.")
125
 
126
  parallelModel = gr.Parallel(generator1,
127
  generator2,
128
  generator3,
129
+ inputs = SplitterInputBox,
130
  examples=examples,
131
  title="Mindfulness Story Generation with Persistent Dataset Memory",
132
  description=f"Mindfulness Story Generation with Persistent Dataset Memory",
 
135
 
136
  tbMemoryOutput = gr.Textbox(label="Memory Output")
137
  btnSave = gr.Button("Save")
138
+ btnSave.click(fn=persist_memory, inputs=[SplitterInputBox, tbOutput], outputs=tbMemoryOutput)
139
 
140
  parallelModel.launch(share=False)