Update README.md
Browse files
README.md
CHANGED
|
@@ -140,12 +140,18 @@ Treat results as **proof-of-concept**, not production-ready.
|
|
| 140 |
|
| 141 |
---
|
| 142 |
|
| 143 |
-
##
|
| 144 |
|
| 145 |
-
|
| 146 |
|
| 147 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 148 |
|
| 149 |
-
## Model Card Contact
|
| 150 |
-
Xinxuan Tang — xinxuant@andrew.cmu.edu
|
| 151 |
|
|
|
|
| 140 |
|
| 141 |
---
|
| 142 |
|
| 143 |
+
## How to Get Started with the Model
|
| 144 |
|
| 145 |
+
You can load this fine-tuned DistilBERT model directly from the Hugging Face Hub and run inference:
|
| 146 |
|
| 147 |
+
```python
|
| 148 |
+
from transformers import pipeline
|
| 149 |
+
|
| 150 |
+
# Load the fine-tuned model
|
| 151 |
+
classifier = pipeline("text-classification", model="YOUR_USERNAME/finetuned_model")
|
| 152 |
+
|
| 153 |
+
# Example input
|
| 154 |
+
sample = "This cozy little spot serves delicious tacos with great service!"
|
| 155 |
+
print(classifier(sample))
|
| 156 |
|
|
|
|
|
|
|
| 157 |
|