Kumarkishalaya commited on
Commit
8de647b
1 Parent(s): 3dd89fe

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -185,11 +185,11 @@ iface = gr.Interface(
185
  gr.Textbox(label="GPT-2 Finetuned Model Output")
186
  ],
187
  examples=[
188
- ["no run, short ball, turning away"],
189
- ["jumps down the track and"],
190
- ["another leg bye call. On the pads"],
191
- ["That stays low, turns in, hits the pads"],
192
- ["goes flat, goes big"]
193
  ],
194
  title="Next word prediction with RNN, LSTM and GPT-2 Models",
195
  description="Start writing a cricket commentary and various models will try to continue it. Compare outputs from LSTM, RNN, and GPT-2 (base & finetuned) models. The RNN & LSTM models both have a single hidden layer trained for 50 epochs, the GPT-2 has been trained for 3 epochs. All models have been trained on 1000 rows of cricket commentary text (~80k word tokens)"
 
185
  gr.Textbox(label="GPT-2 Finetuned Model Output")
186
  ],
187
  examples=[
188
+ ["no run, short ball, turning away", 50, 50, 0.7],
189
+ ["jumps down the track and", 50, 50, 0.7],
190
+ ["another leg bye call. On the pads", 50, 50, 0.7],
191
+ ["That stays low, turns in, hits the pads", 50, 50, 0.7],
192
+ ["goes flat, goes big", 50, 50, 0.7]
193
  ],
194
  title="Next word prediction with RNN, LSTM and GPT-2 Models",
195
  description="Start writing a cricket commentary and various models will try to continue it. Compare outputs from LSTM, RNN, and GPT-2 (base & finetuned) models. The RNN & LSTM models both have a single hidden layer trained for 50 epochs, the GPT-2 has been trained for 3 epochs. All models have been trained on 1000 rows of cricket commentary text (~80k word tokens)"