Update app.py

app.py

@@ -1,5 +1,8 @@
 from flask import Flask, request
 from transformers import RobertaForSequenceClassification, RobertaTokenizer, RobertaConfig
+from transformers import AutoTokenizer, AutoModelForSequenceClassification
+from transformers import RobertaConfig
+from torch import cuda
 import torch
 import gradio as gr
 import os
@@ -9,11 +12,14 @@ import pdfplumber
 app = Flask(__name__)
 
 ACCESS_TOKEN = os.environ["ACCESS_TOKEN"]
-config = RobertaConfig.from_pretrained("PirateXX/ChatGPT-Text-Detector", use_auth_token= ACCESS_TOKEN)
-model = RobertaForSequenceClassification.from_pretrained("PirateXX/ChatGPT-Text-Detector", use_auth_token= ACCESS_TOKEN, config = config)
+# config = RobertaConfig.from_pretrained("PirateXX/ChatGPT-Text-Detector", use_auth_token= ACCESS_TOKEN)
+# model = RobertaForSequenceClassification.from_pretrained("PirateXX/ChatGPT-Text-Detector", use_auth_token= ACCESS_TOKEN, config = config)
+
+# model_name = "roberta-base"
+# tokenizer = RobertaTokenizer.from_pretrained(model_name, map_location=torch.device('cpu'))
+tokenizer = AutoTokenizer.from_pretrained("PirateXX/AI-Content-Detector", use_auth_token= ACCESS_TOKEN)
+model = AutoModelForSequenceClassification.from_pretrained("PirateXX/AI-Content-Detector", use_auth_token= ACCESS_TOKEN)
 
-model_name = "roberta-base"
-tokenizer = RobertaTokenizer.from_pretrained(model_name, map_location=torch.device('cpu'))
 
 # function to break text into an array of sentences
 def text_to_sentences(text):