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)
|