Hugging Face's logo

Spaces:
nppmatt
/
milestone-3
Runtime error

App Files Community
nppmatt commited on
Commit
47c22ce
•
1 Parent(s): b420b00

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -88,7 +88,7 @@ MAX_LENGTH = 100
88
  INFER_BATCH_SIZE = 128
89
  infer_dataset = ToxicityDataset(tweet_df, tokenizer, MAX_LENGTH)
90
  infer_params = {"batch_size": INFER_BATCH_SIZE, "shuffle": False}
91
- infer_loader = DataLoader(infer_dataset, **test_params)
92
 
93
  # Freeze model and input tokens
94
  def inference():
 
88
  INFER_BATCH_SIZE = 128
89
  infer_dataset = ToxicityDataset(tweet_df, tokenizer, MAX_LENGTH)
90
  infer_params = {"batch_size": INFER_BATCH_SIZE, "shuffle": False}
91
+ infer_loader = DataLoader(infer_dataset, **infer_params)
92
 
93
  # Freeze model and input tokens
94
  def inference():