Kumarkishalaya commited on
Commit
99b4c75
·
verified ·
1 Parent(s): 7850720

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -185,7 +185,7 @@ iface = gr.Interface(
185
  gr.Textbox(label="GPT-2 Finetuned Model Output")
186
  ],
187
  title="Next word prediction with RNN, LSTM and GPT-2 Models",
188
- 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."
189
  )
190
 
191
  # Launch the app
 
185
  gr.Textbox(label="GPT-2 Finetuned Model Output")
186
  ],
187
  title="Next word prediction with RNN, LSTM and GPT-2 Models",
188
+ 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)"
189
  )
190
 
191
  # Launch the app