asd

app.py

@@ -53,7 +53,7 @@
 import gradio as gr
 from openai import OpenAI
 import torch
-from transformers import AutoTokenizer, AutoModelForSequenceClassification
+from transformers import AutoTokenizer, AutoModelForSequenceClassification, TFAutoModelForSequenceClassification
 
 # define the openai key
 api_key = "sk-proj-UCoZZMs4MyfyHwXdHjT8T3BlbkFJjYkSZyPfIPNqXfXwoekm"
@@ -65,8 +65,14 @@ client = OpenAI(api_key = api_key)
 finetuned_model = "ft:gpt-3.5-turbo-0125:personal::9qGC8cwZ"
 
 # Load model directly
-tokenizer = AutoTokenizer.from_pretrained("tommyliphys/ai-detector-distilbert")
-model = AutoModelForSequenceClassification.from_pretrained("tommyliphys/ai-detector-distilbert")
+model_name = "tommyliphys/ai-detector-distilbert"
+tokenizer = AutoTokenizer.from_pretrained(model_name)
+
+# Load TensorFlow model
+tf_model = TFAutoModelForSequenceClassification.from_pretrained(model_name, from_tf=True)
+
+# Convert to PyTorch model
+model = AutoModelForSequenceClassification.from_pretrained(model_name, from_tf=True)
 
 # Define the function to get predictions
 def get_prediction(text):