umang018 commited on
Commit
616d1b5
·
verified ·
1 Parent(s): 1426c29

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -11,7 +11,7 @@ dataset = load_dataset("Hellisotherpeople/enron_emails_parsed")
11
  enron_data = pd.DataFrame(dataset['train'])
12
 
13
  # Load the model and tokenizer
14
- model_name = "modelSamLowe/roberta-base-go_emotions"
15
  tokenizer = AutoTokenizer.from_pretrained(model_name)
16
  model = AutoModelForSequenceClassification.from_pretrained(model_name).to(device)
17
 
 
11
  enron_data = pd.DataFrame(dataset['train'])
12
 
13
  # Load the model and tokenizer
14
+ model_name = "SamLowe/roberta-base-go_emotions"
15
  tokenizer = AutoTokenizer.from_pretrained(model_name)
16
  model = AutoModelForSequenceClassification.from_pretrained(model_name).to(device)
17