Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -6,13 +6,7 @@ from setfit import SetFitModel
|
|
6 |
def greet(name):
|
7 |
return "Hello " + name + "!!"
|
8 |
|
9 |
-
|
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)
|