Update app.py
Browse files
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)"
|