awacke1 commited on
Commit
fbc8d88
·
1 Parent(s): 335de64

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -6
app.py CHANGED
@@ -113,18 +113,14 @@ iface = gr.Interface(
113
  #demo = gr.Parallel(greeter_1, greeter_2)
114
 
115
  #generator1 = gr.Interface(lambda name: f"Hello {name}!", inputs="textbox", outputs=gr.Textbox(label="GPT2-Large")).load("huggingface/gpt2-large", api_key=HF_TOKEN)
116
- tbOutput = gr.Textbox(label="GPT Output")
117
- tbMemoryOutput = gr.Textbox(label="Memory Output")
118
 
119
  generator1 = gr.Interface(lambda name: f"Hello {name}!", inputs="textbox", outputs=[tbOutput]).load("huggingface/gpt2-large", api_key=HF_TOKEN)
120
  generator2 = gr.Interface.load("huggingface/EleutherAI/gpt-neo-2.7B", api_key=HF_TOKEN)
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
- btnSave = gr.Button("Save")
125
- btnSave.click(fn=persist_memory, inputs=[tbOutput], outputs=[tbMemoryOutput])
126
 
127
- gr.Parallel(generator1,
128
  generator2,
129
  generator3,
130
  inputs = gr.inputs.Textbox(lines=5, label="Enter a sentence to get another sentence."),
@@ -132,4 +128,11 @@ gr.Parallel(generator1,
132
  title="Mindfulness Story Generation with Persistent Dataset Memory",
133
  description=f"Mindfulness Story Generation with Persistent Dataset Memory",
134
  article=f"Memory Dataset URL: [{DATASET_REPO_URL}]({DATASET_REPO_URL})"
135
- ).launch(share=False)
 
 
 
 
 
 
 
 
113
  #demo = gr.Parallel(greeter_1, greeter_2)
114
 
115
  #generator1 = gr.Interface(lambda name: f"Hello {name}!", inputs="textbox", outputs=gr.Textbox(label="GPT2-Large")).load("huggingface/gpt2-large", api_key=HF_TOKEN)
 
 
116
 
117
  generator1 = gr.Interface(lambda name: f"Hello {name}!", inputs="textbox", outputs=[tbOutput]).load("huggingface/gpt2-large", api_key=HF_TOKEN)
118
  generator2 = gr.Interface.load("huggingface/EleutherAI/gpt-neo-2.7B", api_key=HF_TOKEN)
119
  generator3 = gr.Interface.load("huggingface/EleutherAI/gpt-j-6B", api_key=HF_TOKEN)
120
 
121
  #MemoryChange=tbOutput.change(persist_memory,inputs=[tbOutput],outputs=gr.Textbox(label="PersistMemoryOutput"))
 
 
122
 
123
+ parallelModel = gr.Parallel(generator1,
124
  generator2,
125
  generator3,
126
  inputs = gr.inputs.Textbox(lines=5, label="Enter a sentence to get another sentence."),
 
128
  title="Mindfulness Story Generation with Persistent Dataset Memory",
129
  description=f"Mindfulness Story Generation with Persistent Dataset Memory",
130
  article=f"Memory Dataset URL: [{DATASET_REPO_URL}]({DATASET_REPO_URL})"
131
+ )
132
+
133
+ tbOutput = gr.Textbox(label="GPT Output")
134
+ tbMemoryOutput = gr.Textbox(label="Memory Output")
135
+ btnSave = gr.Button("Save")
136
+ btnSave.click(fn=persist_memory, inputs=tbOutput, outputs=tbMemoryOutput)
137
+
138
+ parallelModel.launch(share=False)