Update app.py

app.py

@@ -1,33 +1,38 @@
 import gradio as gr
-from transformers import pipeline
+from generated_text_detector.utils.text_detector import GeneratedTextDetector
 
-# Инициализация пайплайна для классификации текста
-pipe = pipeline("text-classification", model="SuperAnnotate/ai-detector")
+# Инициализация детектора
+detector = GeneratedTextDetector(
+    "SuperAnnotate/ai-detector",
+    device="cuda",
+    preprocessing=True
+)
 
-def classify_text(input_text):
-    # Получаем предсказание от модели
-    result = pipe(input_text)
-    
-    # Извлекаем метку и вероятность
-    label = result[0]['label']
-    score = result[0]['score']
-    
-    # Форматируем результат
-    if label == 'POSITIVE':
-        interpretation = "Текст похож на сгенерированный ИИ"
-    else:
-        interpretation = "Текст похож на человеческий"
-        
-    return f"Результат: {interpretation}\nВероятность: {score:.2%}"
+# Функция для предсказания
+def detect_generated_text(input_text):
+    try:
+        result = detector.detect_report(input_text)
+        return f"Detection Result:\n\n{result}"
+    except Exception as e:
+        return f"Error: {str(e)}"
 
-# Создаем интерфейс Gradio
+# Создание интерфейса Gradio
 interface = gr.Interface(
-    fn=classify_text,
-    inputs=gr.Textbox(lines=5, placeholder="Введите текст здесь..."),
-    outputs="text",
-    title="Детектор текста ИИ",
-    description="Введите текст, чтобы определить, похож ли он на сгенерированный ИИ или написанный человеком."
+    fn=detect_generated_text,
+    inputs=gr.Textbox(
+        lines=5,
+        placeholder="Enter text here to check if it's AI-generated...",
+        label="Input Text"
+    ),
+    outputs=gr.Textbox(label="Detection Result"),
+    title="AI-Generated Text Detector",
+    description="Enter any text to determine if it was generated by AI. Powered by SuperAnnotate/ai-detector model.",
+    examples=[
+        ["It's not uncommon for people to develop allergies or intolerances to certain foods as they get older."],
+        ["The quick brown fox jumps over the lazy dog near the river bank on a sunny afternoon."]
+    ]
 )
 
-# Запускаем интерфейс
-interface.launch()
+# Запуск интерфейса
+if __name__ == "__main__":
+    interface.launch()