raj-tomar001 commited on
Commit
880a0f2
1 Parent(s): 568b9b5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -2,9 +2,8 @@ import gradio as gr
2
  from transformers import DebertaTokenizer, DebertaForSequenceClassification, DistilBertTokenizer, DistilBertForSequenceClassification
3
  from transformers import pipeline
4
  import json
5
- import numpy as np
6
-
7
-
8
 
9
  save_path_abstract = './fine-tuned-distillberta'
10
  model_abstract = DistilBertForSequenceClassification.from_pretrained(save_path_abstract)
 
2
  from transformers import DebertaTokenizer, DebertaForSequenceClassification, DistilBertTokenizer, DistilBertForSequenceClassification
3
  from transformers import pipeline
4
  import json
5
+ import numpy as np
6
+ import random
 
7
 
8
  save_path_abstract = './fine-tuned-distillberta'
9
  model_abstract = DistilBertForSequenceClassification.from_pretrained(save_path_abstract)