Hugging Face's logo

shirleylqs
/
mistral-snomed-classification

Text Classification
Transformers
Safetensors
mistral
Generated from Trainer
text-generation-inference
Inference Endpoints
Model card Files Community
1
shirleylqs commited on
Commit
80408bd
•
1 Parent(s): 6d54929

Upload model and tokenizer

Browse files
Files changed (1) hide show
  1. predict.py +1 -1
predict.py CHANGED
@@ -4,7 +4,7 @@ import torch
4
  from transformers import AutoTokenizer, AutoModelForSequenceClassification
5
  import pickle
6
 
7
- model_path = 'shirleylqs/mistral7B-snomed-classification'
8
  tokenizer = AutoTokenizer.from_pretrained(model_path)
9
  model = AutoModelForSequenceClassification.from_pretrained(model_path)
10
 
 
4
  from transformers import AutoTokenizer, AutoModelForSequenceClassification
5
  import pickle
6
 
7
+ model_path = 'shirleylqs/mistral-snomed-classification'
8
  tokenizer = AutoTokenizer.from_pretrained(model_path)
9
  model = AutoModelForSequenceClassification.from_pretrained(model_path)
10