aalkaswan commited on
Commit
86ffe77
Β·
1 Parent(s): 0dbab37

Update app.py

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