Update app.py
Browse files
app.py
CHANGED
|
@@ -41,6 +41,11 @@ model_args.evaluate_during_training = True
|
|
| 41 |
model_args.n_best_size = 3
|
| 42 |
model_args.num_train_epochs = 5
|
| 43 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 44 |
# Gradio interface function
|
| 45 |
def chatbot(question):
|
| 46 |
to_predict = [
|
|
|
|
| 41 |
model_args.n_best_size = 3
|
| 42 |
model_args.num_train_epochs = 5
|
| 43 |
|
| 44 |
+
# Model definition
|
| 45 |
+
model = QuestionAnsweringModel('bert', 'bert-base-uncased', args={'overwrite_output_dir': True, 'num_train_epochs': 20})
|
| 46 |
+
model.train_model(adapted_data, num_train_epochs=20)
|
| 47 |
+
model.save_model(f"outputs/bert/final_model")
|
| 48 |
+
|
| 49 |
# Gradio interface function
|
| 50 |
def chatbot(question):
|
| 51 |
to_predict = [
|