Hugging Face's logo

Spaces:
Stanpie3
/
MyPaperClassifier
Sleeping

App Files Community
Stanpie3 commited on
Commit
a5f1ae8
·
verified ·
1 Parent(s): 04ffc3a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -31,7 +31,7 @@ class BERTClass(BertPreTrainedModel):
31
  loss = loss_fct(logits, labels)
32
  return {"loss": loss, "logits": logits}
33
 
34
- MODEL_PATH = "./"
35
  LABELS = ['astro-ph', 'cond-mat', 'cs', 'eess', 'gr-qc',
36
  'hep-ex', 'hep-lat', 'hep-ph', 'hep-th', 'math', 'math-ph', 'nlin',
37
  'nucl-ex', 'nucl-th', 'physics', 'q-bio', 'quant-ph', 'stat']
 
31
  loss = loss_fct(logits, labels)
32
  return {"loss": loss, "logits": logits}
33
 
34
+ MODEL_PATH = "."
35
  LABELS = ['astro-ph', 'cond-mat', 'cs', 'eess', 'gr-qc',
36
  'hep-ex', 'hep-lat', 'hep-ph', 'hep-th', 'math', 'math-ph', 'nlin',
37
  'nucl-ex', 'nucl-th', 'physics', 'q-bio', 'quant-ph', 'stat']