inoid commited on
Commit
a415c7f
1 Parent(s): c6ee43b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -70,7 +70,7 @@ def thesis_prediction(input):
70
  output_hidden_states=False)
71
 
72
  pipe = TextClassificationPipeline(model=model, tokenizer=tokenizer, return_all_scores=True)
73
- classificationResult = pipe(_text)
74
  classificationResult[0].sort(reverse=True, key=lambda x:x['score'])
75
  keyClass = classificationResult[0][0]['label']
76
  # # Return the text clasification
 
70
  output_hidden_states=False)
71
 
72
  pipe = TextClassificationPipeline(model=model, tokenizer=tokenizer, return_all_scores=True)
73
+ classificationResult = pipe(input)
74
  classificationResult[0].sort(reverse=True, key=lambda x:x['score'])
75
  keyClass = classificationResult[0][0]['label']
76
  # # Return the text clasification