Spaces:

AISE-TUDelft
/

STACC

Runtime error

aalkaswan commited on Feb 4, 2023

Commit

8ef424d

1 Parent(s): 0e39988

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -7,7 +7,7 @@ def greet(name):
 model = SetFitModel.from_pretrained("aalkaswan/java-deprecation-classifier",
                                     device='cpu', #Use this if you don't have a GPU
-                                    )
 def classify(text):
     return model(text)

 model = SetFitModel.from_pretrained("aalkaswan/java-deprecation-classifier",
                                     device='cpu', #Use this if you don't have a GPU
+                                    token=os.environ.get('sec_tok'))
 def classify(text):
     return model(text)