Alexvatti commited on
Commit
e2bc41e
·
verified ·
1 Parent(s): c1e3de2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -43,7 +43,7 @@ eval_df = pd.DataFrame({"text": X_test, "labels": y_test})
43
 
44
 
45
  # Create a ClassificationModel
46
- model = ClassificationModel("bert", "bert-base-uncased", use_cuda=True) # Set use_cuda=True if you have a GPU
47
 
48
  # Train the model
49
  model.train_model(train_df)
 
43
 
44
 
45
  # Create a ClassificationModel
46
+ model = ClassificationModel("bert", "bert-base-uncased", use_cuda=False) # Set use_cuda=True if you have a GPU
47
 
48
  # Train the model
49
  model.train_model(train_df)