VeenaGarre commited on
Commit
d317940
·
verified ·
1 Parent(s): 708d28a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -20,7 +20,7 @@ if device.type == 'cuda':
20
  import pandas as pd
21
  from datasets import load_dataset
22
  import numpy as np
23
- review_dataset = load_dataset("McAuley-Lab/Amazon-Reviews-2023","raw_review_All_Beauty")
24
 
25
  # User Reviews dataframe (reviews are in the 'train' split)
26
  reviews_df = pd.DataFrame(review_dataset['full'])
 
20
  import pandas as pd
21
  from datasets import load_dataset
22
  import numpy as np
23
+ review_dataset = load_dataset("McAuley-Lab/Amazon-Reviews-2023","raw_review_All_Beauty",trust_remote_code=True)
24
 
25
  # User Reviews dataframe (reviews are in the 'train' split)
26
  reviews_df = pd.DataFrame(review_dataset['full'])