Nandoguer commited on
Commit
3962dbd
·
verified ·
1 Parent(s): d0c0ec4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -13,8 +13,8 @@ from sklearn.model_selection import train_test_split # Import train_test_split
13
  import gradio as gr
14
 
15
  # Load the datasets for emotion detection and quotes
16
- emotion_dataset = load_dataset("dair-ai/emotion")
17
- quotes_dataset = load_dataset("Abirate/english_quotes")
18
 
19
  # Prepare the Bag-of-Words Model
20
  vectorizer = CountVectorizer()
 
13
  import gradio as gr
14
 
15
  # Load the datasets for emotion detection and quotes
16
+ emotion_dataset = load_dataset("dair-ai/emotion",trust_remote_code=True)
17
+ quotes_dataset = load_dataset("Abirate/english_quotes",trust_remote_code=True)
18
 
19
  # Prepare the Bag-of-Words Model
20
  vectorizer = CountVectorizer()