This commit is contained in:
Maciej Ścigacz 2023-05-30 15:47:36 +02:00
parent 6102ba8889
commit 4e9db47859

View File

@ -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):