Transformers How to use rishikasrinivas/bert-genre-pred with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("text-classification", model="rishikasrinivas/bert-genre-pred") # Load model directly
from transformers import AutoModel
model = AutoModel.from_pretrained("rishikasrinivas/bert-genre-pred", device_map="auto") 
