asd

app.py

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