Dominick Meady commited on
Commit
2041000
1 Parent(s): 85ca7d2

changed lib

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -2,8 +2,8 @@ import gradio as gr
2
  from transformers import AutoTokenizer, AutoModelForSequenceClassification
3
 
4
  def plagiarism(input):
5
- tokenizer = AutoTokenizer.from_pretrained("jpelhaw/xlnet-base-plagiarism-detection")
6
- model = AutoModelForSequenceClassification.from_pretrained("jpelhaw/xlnet-base-plagiarism-detection")
7
  example = tokenizer.tokenize(input, add_special_tokens=True)
8
 
9
  answer = model(**example)
 
2
  from transformers import AutoTokenizer, AutoModelForSequenceClassification
3
 
4
  def plagiarism(input):
5
+ tokenizer = AutoTokenizer.from_pretrained("jpelhaw/longformer-base-plagiarism-detection")
6
+ model = AutoModelForSequenceClassification.from_pretrained("jpelhaw/longformer-base-plagiarism-detection")
7
  example = tokenizer.tokenize(input, add_special_tokens=True)
8
 
9
  answer = model(**example)