Hugging Face's logo

Spaces:
AISE-TUDelft
/
STACC
Runtime error

App Files Community
aalkaswan commited on
Commit
859d030
·
1 Parent(s): 1cd80b7

Update app.py

Browse files

Update to use V1 of models

Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -8,7 +8,7 @@ model_names = ['java-summary', 'java-pointer', 'java-deprecation', 'java-rationa
8
 
9
  models = {}
10
  for model_name in model_names:
11
- models[model_name] = SetFitModel.from_pretrained(f'AISE-TUDelft/{model_name}-classifier')#
12
 
13
 
14
  def classify(text, model_name):
 
8
 
9
  models = {}
10
  for model_name in model_names:
11
+ models[model_name] = SetFitModel.from_pretrained(f'AISE-TUDelft/{model_name}-classifier', revision='V1')#
12
 
13
 
14
  def classify(text, model_name):