AliHaider0343 commited on
Commit
eba1c32
·
1 Parent(s): c1afd53

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -1,12 +1,12 @@
1
  import torch
2
  import nltk
3
  import streamlit as st
4
- from nltk.corpus import stopwords
5
  from transformers import RobertaTokenizer, RobertaForSequenceClassification
 
 
6
  stop_words = set(stopwords.words('english'))
7
  stop_words.discard('and')
8
 
9
-
10
  def tokenize_sentences(sentence):
11
  encoded_dict = tokenizer.encode_plus(
12
  sentence,
 
1
  import torch
2
  import nltk
3
  import streamlit as st
 
4
  from transformers import RobertaTokenizer, RobertaForSequenceClassification
5
+ nltk.download('stopwords')
6
+ nltk.download('punkt')
7
  stop_words = set(stopwords.words('english'))
8
  stop_words.discard('and')
9
 
 
10
  def tokenize_sentences(sentence):
11
  encoded_dict = tokenizer.encode_plus(
12
  sentence,