test
This commit is contained in:
parent
6102ba8889
commit
4e9db47859
@ -12,7 +12,6 @@ tokenizer = AutoTokenizer.from_pretrained("Scigi/sentiment-analysis-model")
|
|||||||
def sentiment_prediction(data):
|
def sentiment_prediction(data):
|
||||||
pipe = pipeline('text-classification', model=model, tokenizer = tokenizer)
|
pipe = pipeline('text-classification', model=model, tokenizer = tokenizer)
|
||||||
result = pipe(data)
|
result = pipe(data)
|
||||||
|
|
||||||
return result
|
return result
|
||||||
|
|
||||||
def clear_data(data):
|
def clear_data(data):
|
||||||
|
Loading…
Reference in New Issue
Block a user