chinmaygarde
/

hello

Text Classification

sentiment-analysis

Model card Files Files and versions

chinmaygarde commited on 30 days ago

Commit

9cacd9e

·

unverified ·

1 Parent(s): 9759036

Update model.

Files changed (2) hide show

hello.py +1 -1
model.safetensors +1 -1

hello.py CHANGED Viewed

@@ -7,7 +7,7 @@ from transformers import (
 from datasets import load_dataset
 # Load a small subset of IMDB reviews
-dataset = load_dataset("imdb", split="train[:500]")
 dataset = dataset.train_test_split(test_size=0.2, seed=42)
 # Use DistilBERT — small, fast, good enough for a demo

 from datasets import load_dataset
 # Load a small subset of IMDB reviews
+dataset = load_dataset("stanfordnlp/imdb", split="train[:500]")
 dataset = dataset.train_test_split(test_size=0.2, seed=42)
 # Use DistilBERT — small, fast, good enough for a demo

model.safetensors CHANGED Viewed

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:bbc6af1fce00266a50502c0f51837e9a4b7bee649978e8a7ed86db4764b47e64
 size 267832560

 version https://git-lfs.github.com/spec/v1
+oid sha256:7bae3d46f16ba7a391ca550d66f54205f643c845b661c0f603009c98d7f828c1
 size 267832560