aalkaswan commited on
Commit
27f7e99
Β·
1 Parent(s): fb9083d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -8,7 +8,7 @@ def greet(name):
8
 
9
  model = SetFitModel.from_pretrained("AISE-TUDelft/java-ownership-classifier",
10
  device='cpu', #Use this if you don't have a GPU
11
- token=os.environ.get('sec_tok'))
12
 
13
  def classify(text):
14
  return model(text)
 
8
 
9
  model = SetFitModel.from_pretrained("AISE-TUDelft/java-ownership-classifier",
10
  device='cpu', #Use this if you don't have a GPU
11
+ use_auth_token=os.environ.get('sec_tok'))
12
 
13
  def classify(text):
14
  return model(text)