awacke1 commited on
Commit
e1f5731
·
1 Parent(s): 0f8469c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -124,7 +124,7 @@ tbOutput = gr.Textbox(label="GPT Output")
124
  #model_1_iface = gr.Interface( fn=your_function_1, inputs=gr.inputs.Textbox(), outputs=gr.outputs.Label(num_top_classes=10))
125
  #model_2_iface = gr.Interface( fn= your_function_2, inputs=gr.inputs.Textbox(), outputs=gr.outputs.Label(num_top_classes=10),)
126
 
127
- generator1 = gr.Interface.load(fn=persist_memory, "huggingface/gpt2-large", api_key=HF_TOKEN, inputs=gr.inputs.Textbox(), outputs=gr.outputs.Label(num_top_classes=10))
128
  generator2 = gr.Interface.load("huggingface/EleutherAI/gpt-neo-2.7B", api_key=HF_TOKEN)
129
  generator3 = gr.Interface.load("huggingface/EleutherAI/gpt-j-6B", api_key=HF_TOKEN)
130
 
 
124
  #model_1_iface = gr.Interface( fn=your_function_1, inputs=gr.inputs.Textbox(), outputs=gr.outputs.Label(num_top_classes=10))
125
  #model_2_iface = gr.Interface( fn= your_function_2, inputs=gr.inputs.Textbox(), outputs=gr.outputs.Label(num_top_classes=10),)
126
 
127
+ generator1 = gr.Interface(fn=persist_memory, inputs=gr.inputs.Textbox(), outputs=gr.outputs.Label(num_top_classes=10) ).load("huggingface/gpt2-large", api_key=HF_TOKEN)
128
  generator2 = gr.Interface.load("huggingface/EleutherAI/gpt-neo-2.7B", api_key=HF_TOKEN)
129
  generator3 = gr.Interface.load("huggingface/EleutherAI/gpt-j-6B", api_key=HF_TOKEN)
130