Update app.py

app.py

@@ -1,21 +1,20 @@
 import gradio as gr
 import torch
-from moderation import *  # Убедитесь, что в moderation.py есть функции getEmb и predict
+from moderation import *
+
 
-# Загрузка модели
 moderation = ModerationModel()
-moderation.load_state_dict(torch.load('moderation_model.pth', map_location=torch.device('cpu')))
-moderation.eval()  # Переключение модели в режим оценки
+moderation.load_state_dict(torch.load('moderation_model.pth', map_location=torch.device('cpu'))) #Remove map_location if run on gpu
+moderation.eval()
 
 def predict_moderation(text):
     embeddings_for_prediction = getEmb(text)
     prediction = predict(moderation, embeddings_for_prediction)
-    # Предполагая, что prediction возвращает словарь с оценками и флагом обнаружения
-    category_scores = prediction.get('category_scores', {})  # Извлечение оценок категорий из словаря
-    detected = prediction.get('detected', False)  # Извлечение флага обнаружения
-    return category_scores, str(detected)  # Преобразование detected в строку для отображения
+    category_scores = prediction.get('category_scores', {})
+    detected = prediction.get('detected', False)
+    return category_scores, str(detected)
+
 
-# Создание интерфейса Gradio
 iface = gr.Interface(fn=predict_moderation,
                      inputs="text",
                      outputs=[gr.outputs.Label(label="Category Scores", type="confidences"),
@@ -23,5 +22,5 @@ iface = gr.Interface(fn=predict_moderation,
                      title="Moderation Model",
                      description="Enter text to check for moderation flags.")
 
-# Запуск интерфейса
+
 iface.launch()