Hugging Face's logo

Spaces:
Erfan11
/
Textwizai
Running

App Files Community
1
Erfan11 commited on
Commit
06f46aa
•
1 Parent(s): f536b77

Create load_model.py

Browse files
Files changed (1) hide show
  1. load_model.py +3 -6
load_model.py CHANGED
@@ -1,7 +1,4 @@
1
- import torch
2
- from transformers import AutoModelForSequenceClassification
3
 
4
- def load_model(model_path):
5
- model = AutoModelForSequenceClassification.from_pretrained(model_path)
6
- model.eval()
7
- return model
 
1
+ from transformers import BertForSequenceClassification
 
2
 
3
+ # Load the model from your Hugging Face repository
4
+ model = BertForSequenceClassification.from_pretrained("Erfan11/Neuracraft", from_tf=True)