asd

app.py

@@ -54,7 +54,7 @@ import gradio as gr
 from openai import OpenAI
 import os
 import re
-from transformers import pipeline, DistilBertForSequenceClassification, DistilBertTokenizerFast
+from transformers import pipeline
 
 # define the openai key
 api_key = "sk-proj-UCoZZMs4MyfyHwXdHjT8T3BlbkFJjYkSZyPfIPNqXfXwoekm"
@@ -66,11 +66,7 @@ client = OpenAI(api_key = api_key)
 finetuned_model = "ft:gpt-3.5-turbo-0125:personal::9qGC8cwZ"
 
 # Load the AI detection model
-model_name = "tommyliphys/ai-detector-distilbert"
-model = DistilBertForSequenceClassification.from_pretrained(model_name)
-tokenizer = DistilBertTokenizerFast.from_pretrained(model_name)
-
-pipe = pipeline("text-classification", model=model, tokenizer=tokenizer)
+pipe = pipeline("text-classification", model="tommyliphys/ai-detector-distilbert")
 
 # Define the function to get predictions
 def get_prediction(text):