Rajut commited on
Commit
755f869
·
verified ·
1 Parent(s): 60c291e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -42,11 +42,12 @@ 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, 'use_cuda': False})
46
  model.train_model(adapted_data, num_train_epochs=20)
47
  model.save_model(f"outputs/bert/final_model")
48
 
49
 
 
50
  # Gradio interface function
51
  def chatbot(question):
52
  to_predict = [
 
42
  model_args.num_train_epochs = 5
43
 
44
  # Model definition
45
+ model = QuestionAnsweringModel('bert', 'bert-base-uncased', use_cuda=False, 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
 
50
+
51
  # Gradio interface function
52
  def chatbot(question):
53
  to_predict = [