Spaces:

AISE-TUDelft
/

STACC

Runtime error

aalkaswan commited on Feb 15, 2023

Commit

a3ed488

1 Parent(s): 859d030

Update app.py

Change revision to tag

Files changed (1) hide show

app.py CHANGED Viewed

@@ -8,7 +8,7 @@ model_names = ['java-summary', 'java-pointer', 'java-deprecation', 'java-rationa
 models = {}
 for model_name in model_names:
-    models[model_name] = SetFitModel.from_pretrained(f'AISE-TUDelft/{model_name}-classifier', revision='V1')#
 def classify(text, model_name):

 models = {}
 for model_name in model_names:
+    models[model_name] = SetFitModel.from_pretrained(f'AISE-TUDelft/{model_name}-classifier', tag='V1')#
 def classify(text, model_name):