model_upload / app.py
Pavankumar0619's picture
Create app.py
a48e945
raw
history blame
360 Bytes
from transformers import AutoModelForSequenceClassification, AutoTokenizer
model_name = "intfloat/e5-mistral-7b-instruct" # Use the correct identifier for your model
tokenizer = AutoTokenizer.from_pretrained(model_name)
model = AutoModelForSequenceClassification.from_pretrained(model_name)
# Now, you can use the 'tokenizer' and 'model' for your NLP tasks