Rajut commited on
Commit
6ffe34a
·
verified ·
1 Parent(s): 308551f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -0
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 = [