Rajut commited on
Commit
af030cc
·
verified ·
1 Parent(s): ef94f94

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -42,8 +42,8 @@ model_args.n_best_size = 3
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': 4})
46
- model.train_model(adapted_data, num_train_epochs=4)
47
  model.save_model(f"outputs/bert/final_model")
48
 
49
 
@@ -77,4 +77,4 @@ iface = gr.Interface(
77
  )
78
 
79
  # Launch Gradio interface
80
- iface.launch()
 
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': 6})
46
+ model.train_model(adapted_data, num_train_epochs=6)
47
  model.save_model(f"outputs/bert/final_model")
48
 
49
 
 
77
  )
78
 
79
  # Launch Gradio interface
80
+ iface.launch(share=True)