Update README.md

README.md

@@ -1,3 +1,6 @@
+---
+pipeline_tag: text-classification
+---
     tokenizer = BartTokenizer.from_pretrained('ihgn/paraphrase-detection')
     model = BartForConditionalGeneration.from_pretrained("ihgn/paraphrase-detection").to(device)
     source_sentence = "This was a series of nested angular standards , so that measurements in azimuth and elevation could be done directly in polar coordinates relative to the ecliptic."
@@ -15,4 +18,4 @@
       predicted_label = 1 if generated_text == '1' else 0
       print("Predicted Label:", predicted_label)
 
-    paraphrase_detection(model, tokenizer, source_sentence, target_paraphrase)
+    paraphrase_detection(model, tokenizer, source_sentence, target_paraphrase)