Update app.py

app.py

@@ -13,6 +13,7 @@ from torchvision.models.detection import fasterrcnn_resnet50_fpn
 from torchvision.transforms import functional as F
 import tempfile
 import time
+import requests
 
 # Инициализация клиента для модели HuggingFaceH4/zephyr-7b-beta
 client = InferenceClient("HuggingFaceH4/zephyr-7b-beta")
@@ -21,20 +22,35 @@ client = InferenceClient("HuggingFaceH4/zephyr-7b-beta")
 model = fasterrcnn_resnet50_fpn(pretrained=True)
 model.eval()
 
-# Инициализация модели для определения NSFW
-nsfw_client = InferenceClient("Falconsai/nsfw_image_detection")
+# API для модели NSFW
+API_URL = "https://api-inference.huggingface.co/models/Falconsai/nsfw_image_detection"
+headers = {"Authorization": "Bearer hf_xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx"}  # Замените на ваш API-токен
 
 # Функция для определения NSFW в изображении
 def detect_nsfw(image):
     if image is None:
         return "Изображение не загружено."
     
-    # Отправка изображения в модель NSFW
-    result = nsfw_client.image_to_image(image=image)
+    # Сохраняем изображение во временный файл
+    with tempfile.NamedTemporaryFile(suffix=".jpg", delete=False) as tmp_file:
+        image.save(tmp_file.name)
+        filename = tmp_file.name
     
-    # Получение процента NSFW
-    nsfw_score = result.get("nsfw_score", 0)
-    return f"Процент NSFW: {nsfw_score * 100:.2f}%"
+    # Отправка изображения в модель NSFW через API
+    try:
+        with open(filename, "rb") as f:
+            data = f.read()
+        response = requests.post(API_URL, headers=headers, data=data)
+        result = response.json()
+        
+        # Обработка результата
+        if isinstance(result, list) and len(result) > 0 and "score" in result[0]:
+            nsfw_score = result[0]["score"]
+            return f"Процент NSFW: {nsfw_score * 100:.2f}%"
+        else:
+            return "Не удалось определить NSFW. Попробуйте другое изображение."
+    except Exception as e:
+        return f"Ошибка при обработке изображения: {str(e)}"
 
 # Функция для обработки PDF
 def process_pdf(file):