Create main.py

main.py

@@ -0,0 +1,33 @@
+import gradio as gr
+from transformers import pipeline
+
+# Инициализация пайплайна для классификации текста
+pipe = pipeline("text-classification", model="SuperAnnotate/ai-detector")
+
+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%}"
+
+# Создаем интерфейс Gradio
+interface = gr.Interface(
+    fn=classify_text,
+    inputs=gr.Textbox(lines=5, placeholder="Введите текст здесь..."),
+    outputs="text",
+    title="Детектор текста ИИ",
+    description="Введите текст, чтобы определить, похож ли он на сгенерированный ИИ или написанный человеком."
+)
+
+# Запускаем интерфейс
+interface.launch()