Update app.py

app.py

@@ -2,9 +2,8 @@ import gradio as gr
 from transformers import DebertaTokenizer, DebertaForSequenceClassification, DistilBertTokenizer, DistilBertForSequenceClassification
 from transformers import pipeline
 import json
-import numpy as np 
-
-
+import numpy as np
+import random
 
 save_path_abstract = './fine-tuned-distillberta'
 model_abstract = DistilBertForSequenceClassification.from_pretrained(save_path_abstract)