Update README.md
Browse files
README.md
CHANGED
@@ -35,14 +35,13 @@ This repository contains a fine-tuned DistilBERT model for sentiment classificat
|
|
35 |
Below is an example of how to load and use the model for sentiment classification:
|
36 |
|
37 |
```python
|
38 |
-
from transformers import DistilBertTokenizer,DistilBertForSequenceClassification
|
39 |
import torch
|
40 |
-
import streamlit as st
|
41 |
|
42 |
# Load the tokenizer and model
|
43 |
-
|
44 |
"ashish-001/DistilBert-Amazon-review-sentiment-classifier")
|
45 |
-
|
46 |
"ashish-001/DistilBert-Amazon-review-sentiment-classifier")
|
47 |
|
48 |
# Example usage
|
|
|
35 |
Below is an example of how to load and use the model for sentiment classification:
|
36 |
|
37 |
```python
|
38 |
+
from transformers import DistilBertTokenizer,DistilBertForSequenceClassification
|
39 |
import torch
|
|
|
40 |
|
41 |
# Load the tokenizer and model
|
42 |
+
model = DistilBertForSequenceClassification.from_pretrained(
|
43 |
"ashish-001/DistilBert-Amazon-review-sentiment-classifier")
|
44 |
+
tokenizer = DistilBertTokenizer.from_pretrained(
|
45 |
"ashish-001/DistilBert-Amazon-review-sentiment-classifier")
|
46 |
|
47 |
# Example usage
|