Amenallah2001 commited on
Commit
51edcbb
·
verified ·
1 Parent(s): 871eeeb

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -3
README.md CHANGED
@@ -7,9 +7,7 @@ language:
7
  library_name: transformers
8
  pipeline_tag: text-classification
9
  ---
10
- Creating a model card for your fine-tuned BERT model on Hugging Face involves clearly documenting the purpose, datasets, usage, and other relevant information. Below is an example template for your model card:
11
 
12
- ---
13
 
14
  ## **Model Overview**
15
  This model is a fine-tuned version of BERT designed to classify SMS messages as either spam or not spam. It was developed as part of a technical test for the startup **IntiGo**.
@@ -44,7 +42,7 @@ You can use this model to classify SMS messages into spam or not spam. The model
44
  from transformers import AutoTokenizer, AutoModelForSequenceClassification
45
 
46
  # Load the model and tokenizer
47
- model_name = "your-model-name"
48
  tokenizer = AutoTokenizer.from_pretrained(model_name)
49
  model = AutoModelForSequenceClassification.from_pretrained(model_name)
50
 
 
7
  library_name: transformers
8
  pipeline_tag: text-classification
9
  ---
 
10
 
 
11
 
12
  ## **Model Overview**
13
  This model is a fine-tuned version of BERT designed to classify SMS messages as either spam or not spam. It was developed as part of a technical test for the startup **IntiGo**.
 
42
  from transformers import AutoTokenizer, AutoModelForSequenceClassification
43
 
44
  # Load the model and tokenizer
45
+ model_name = "Amenallah2001/intigo-technical-test"
46
  tokenizer = AutoTokenizer.from_pretrained(model_name)
47
  model = AutoModelForSequenceClassification.from_pretrained(model_name)
48