Dominick Meady commited on
Commit
01e798e
·
1 Parent(s): 2b9940a

updated plagiarism

Browse files
Files changed (1) hide show
  1. app.py +9 -3
app.py CHANGED
@@ -1,7 +1,13 @@
1
  import gradio as gr
 
2
 
3
- def greet(name):
4
- return "Hello " + name + "!!"
 
 
5
 
6
- iface = gr.Interface(fn=greet, inputs="text", outputs="text")
 
 
 
7
  iface.launch()
 
1
  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)
10
+ return answer
11
+
12
+ iface = gr.Interface(fn=plagiarism, inputs="text", outputs="text")
13
  iface.launch()