asd

app.py

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