Update app.py

app.py

@@ -1,9 +1,13 @@
 import gradio as gr
 from transformers import AutoTokenizer, AutoModelForTokenClassification,RobertaTokenizer
 import torch
+import nltk
+from nltk.tokenize import sent_tokenize
 from fin_readability_sustainability import BERTClass, do_predict
 import pandas as pd
 
+nltk.download('punkt')
+
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
 
@@ -13,7 +17,6 @@ model_sustain.to(device)
 model_sustain.load_state_dict(torch.load('sustainability_model.bin', map_location=device)['model_state_dict'])
 
 
-from nltk.tokenize import sent_tokenize
 def get_sustainability(text):
   df = pd.DataFrame({'sentence':sent_tokenize(text)})
   actual_predictions_sustainability = do_predict(model_sustain, tokenizer_sus, df)