ashish-001 commited on
Commit
df7120d
·
verified ·
1 Parent(s): 7678b5d

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -4
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
- tokenizer = DistilBertForSequenceClassification.from_pretrained(
44
  "ashish-001/DistilBert-Amazon-review-sentiment-classifier")
45
- model = DistilBertTokenizer.from_pretrained(
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