plagiarism / app.py
Dominick Meady
changed lib
2041000
raw
history blame
521 Bytes
import gradio as gr
from transformers import AutoTokenizer, AutoModelForSequenceClassification
def plagiarism(input):
tokenizer = AutoTokenizer.from_pretrained("jpelhaw/longformer-base-plagiarism-detection")
model = AutoModelForSequenceClassification.from_pretrained("jpelhaw/longformer-base-plagiarism-detection")
example = tokenizer.tokenize(input, add_special_tokens=True)
answer = model(**example)
return answer
iface = gr.Interface(fn=plagiarism, inputs="text", outputs="text")
iface.launch()