AISimplyExplained commited on
Commit
bc21776
·
verified ·
1 Parent(s): 14c8502

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -50,7 +50,7 @@ app = FastAPI()
50
  guardrail = Guardrail()
51
 
52
 
53
- @app.post("/classify/", response_model=ClassificationResult)
54
  def classify_text(text_prompt: TextPrompt):
55
  try:
56
  result = guardrail.guard(text_prompt.prompt)
 
50
  guardrail = Guardrail()
51
 
52
 
53
+ @app.post("/api/models/PromptInjection/classify", response_model=ClassificationResult)
54
  def classify_text(text_prompt: TextPrompt):
55
  try:
56
  result = guardrail.guard(text_prompt.prompt)