Switch to SuperAnnotate/ai-detector for better accuracy

app.py

@@ -1,9 +1,13 @@
 from fastapi import FastAPI
-from transformers import pipeline
+from transformers import AutoTokenizer, AutoModelForSequenceClassification, pipeline
 from pydantic import BaseModel
+import torch
 
 app = FastAPI()
-detector = pipeline("text-classification", model="roberta-base-openai-detector")
+model_name = "SuperAnnotate/ai-detector"
+tokenizer = AutoTokenizer.from_pretrained(model_name)
+model = AutoModelForSequenceClassification.from_pretrained(model_name)
+detector = pipeline("text-classification", model=model, tokenizer=tokenizer)
 
 class TextInput(BaseModel):
     text: str