Update README.md

README.md

@@ -98,17 +98,16 @@ predict("DVAJ-548_CH_SD")
 
 ## How to Get Started with the Model
 
-### step 1:  
-Save all the model files into a directory, such as 'model'
-### step 2: 
+
+### step 1: 
 Create a python file under this model, such as 'use_model.py'
 ```python
 import torch
 from transformers import BertForSequenceClassification, BertTokenizer
+
 # load model
-model = BertForSequenceClassification.from_pretrained("./sexual_detection")
-# load tokenizer
-tokenizer = BertTokenizer.from_pretrained("./sexual_detection")
+tokenizer = BertTokenizer.from_pretrained("uget/sexual_content_dection")
+model = BertForSequenceClassification.from_pretrained("uget/sexual_content_dection")
 
 def predict(text):
     encoding = tokenizer(text, return_tensors="pt")
@@ -126,7 +125,7 @@ def predict(text):
 predict("Tiffany Doll - Wine Makes Me Anal (31.03.2018)_1080p.mp4")
 
 ```
-### step 3:
+### step 2:
 Run
 ```shell
 python3 use_model.py