Views
No views yet
1from transformers import pipeline
2
3clf = pipeline("text-classification", model="BhuviMohan/distilbert-imdb-finetuned")
4print(clf("The movie was absolutely fantastic!"))
5
6## Training Procedure
7
8- Epochs: 1
9
10- Batch size: 8
11
12- Learning rate: 2e-5
13
14- Optimizer: AdamW
15
16- Hardware: CPU training (Windows)
17
18- Results (Evaluation)
19
20**Accuracy: ~1.0 (on 2k subset)**
21
22
23Save the file.
24
25---
26
27# ✅ **STEP 5 — Upload model to Hugging Face Hub**
28
29Make sure your token is created with **Write access**.
30
31Then login (if not logged in):
32
33```bash
34huggingface-cli login