Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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(
|
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
|