cleanup

app.py

@@ -298,28 +298,12 @@ def audio_classifier(audio, model_name):
 def audio_zero_shot_classifier(audio, candidate_labels, model_name):
     """Zero-shot классификация аудио"""
     try:
-        labels = [label.strip() for label in candidate_labels.split(",")]
         classifier = get_pipeline("zero-shot-audio-classification", model_name)
+        labels = [label.strip() for label in candidate_labels.split(",")]
         result = classifier(audio, candidate_labels=labels)
-        
-        # Обрабатываем результат
-        if isinstance(result, list):
-            result = result[0] if result else {}
-        
         output = "Результаты классификации:\n"
-        if isinstance(result, dict) and 'scores' in result:
-            for label, score in zip(result.get('labels', labels), result['scores']):
-                output += f"{label}: {score:.4f}\n"
-        elif isinstance(result, list):
-            for item in result:
-                if isinstance(item, dict):
-                    label = item.get('label', '')
-                    score = item.get('score', 0.0)
-                    output += f"{label}: {score:.4f}\n"
-        else:
-            # Если формат неожиданный, пытаемся извлечь информацию
-            output += str(result)
-        
+        for label, score in zip(result['labels'], result['scores']):
+            output += f"{label}: {score:.4f}\n"
         return output
     except Exception as e:
         return f"Ошибка: {str(e)}"
@@ -712,26 +696,12 @@ def visual_qa(image, question, model_name):
 def image_zero_shot_classification(image, candidate_labels, model_name):
     """Zero-shot классификация изображений"""
     try:
-        labels = [label.strip() for label in candidate_labels.split(",")]
         classifier = get_pipeline("zero-shot-image-classification", model_name)
+        labels = [label.strip() for label in candidate_labels.split(",")]
         result = classifier(image, candidate_labels=labels)
-        
-        # Обрабатываем результат
-        if isinstance(result, list):
-            # Сортируем по score если нужно
-            result = sorted(result, key=lambda x: x.get('score', 0), reverse=True)
-        else:
-            result = [result] if result else []
-        
         output = "Результаты классификации:\n"
-        for item in result:
-            if isinstance(item, dict):
-                label = item.get('label', '')
-                score = item.get('score', 0.0)
-                output += f"{label}: {score:.4f}\n"
-            else:
-                output += f"{item}\n"
-        
+        for label, score in zip(result['labels'], result['scores']):
+            output += f"{label}: {score:.4f}\n"
         return output
     except Exception as e:
         return f"Ошибка: {str(e)}"