Update README.md

README.md

@@ -7,7 +7,7 @@ tags:
 - imdb
 - tinybert
 datasets:
-- imdb
+- jamander/Blockbuster
 model_type: bert
 base_model: prajjwal1/bert-tiny
 fine_tuned_by: Jack Mander
@@ -126,5 +126,4 @@ This project was completed to explore the effectiveness of parameter-efficient f
      inputs = tokenizer("I loved the movie!", return_tensors="pt", padding=True, truncation=True)
      outputs = model(**inputs)
      prediction = torch.argmax(outputs.logits, dim=-1)
-     print("Prediction:", "Positive" if prediction == 1 else "Negative")
-     
+     print("Prediction:", "Positive" if prediction == 1 else "Negative")