Spaces:

Alanhau
/

yolov7-test

Runtime error

yolov7-test / model-test.py

Update model-test.py

e695e6e over 1 year ago

No virus

383 Bytes

	from transformers import AutoTokenizer, AutoModelForSequenceClassification
	from transformers import pipeline


	tokenizer = AutoTokenizer.from_pretrained("Alanhau/test")
	model = AutoModelForSequenceClassification.from_pretrained("Alanhau/test")

	nlp = pipeline("sentiment-analysis", model=model, tokenizer=tokenizer)

	sentences = ["hate you!"]
	results = nlp(sentences)
	print(results)