Add 1D sentence plots

Change arXiv category in README
llm-security-research · Mar 15, 2024 · 194907a · 194907a
1 parent 236664b
commit 194907a
Show file tree

Hide file tree

Showing 5 changed files with 80 additions and 79 deletions.
diff --git a/README.md b/README.md
@@ -94,6 +94,6 @@ If you find our work useful, please [cite our paper](https://github.com/llm-secu
       year={2024},
       eprint={},
       archivePrefix={arXiv},
-      primaryClass={cs.AI}
+      primaryClass={cs.CL}
 }
 ```
diff --git a/embeddings.pkl b/embeddings.pkl
diff --git a/hubconf.py b/hubconf.py
@@ -33,7 +33,7 @@ def prediction_pipeline(*args, **kwargs):
 
     print("Creating prediction pipeline...")
     pipeline = PredictionPipeline(model)
-    print('Pipeline created successfully. Use PredictionPipeline.predict("{text}") to classify prompts.')
+    print('Pipeline created successfully. Use PredictionPipeline.predict(text) to classify prompts.')
 
     return pipeline