AliHaider0343 commited on
Commit
175a254
1 Parent(s): 0652585

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -4,6 +4,8 @@ from transformers import RobertaTokenizer, RobertaForSequenceClassification
4
  import nltk
5
  from nltk.corpus import stopwords
6
  import re
 
 
7
  nltk.download('stopwords')
8
  nltk.download('punkt')
9
  stop_words = set(stopwords.words('english'))
@@ -34,6 +36,7 @@ def preprocess_query(query):
34
  query = str(query).lower()
35
  query = query.strip()
36
  query = remove_stop_words(query)
 
37
  return query
38
 
39
  def predict_aspects(sentence, threshold):
 
4
  import nltk
5
  from nltk.corpus import stopwords
6
  import re
7
+ import string
8
+
9
  nltk.download('stopwords')
10
  nltk.download('punkt')
11
  stop_words = set(stopwords.words('english'))
 
36
  query = str(query).lower()
37
  query = query.strip()
38
  query = remove_stop_words(query)
39
+ query=query.translate(str.maketrans("", "", string.punctuation))
40
  return query
41
 
42
  def predict_aspects(sentence, threshold):