Update app.py

app.py

@@ -4,6 +4,7 @@ from PIL import Image
 import gradio as gr
 import logging
 from functools import lru_cache
+import os
 
 # Настройка логирования
 logging.basicConfig(level=logging.INFO)
@@ -79,6 +80,23 @@ def classify_image(image):
             "available_labels": list(model.config.id2label.values())[:10] + ["..."]
         }
 
+# Проверка существования файлов примеров
+def check_example_files(example_list):
+    valid_examples = []
+    for example in example_list:
+        if os.path.exists(example[0]):
+            valid_examples.append(example)
+        else:
+            logger.warning(f"Example file not found: {example[0]}")
+    return valid_examples
+
+# Примеры изображений (убедитесь, что файлы существуют)
+example_files = [
+    ["salmon.jpg"],
+    ["trout.jpg"]
+]
+valid_examples = check_example_files(example_files)
+
 # Создание интерфейса Gradio
 iface = gr.Interface(
     fn=classify_image,
@@ -86,10 +104,7 @@ iface = gr.Interface(
     outputs=gr.JSON(label="Classification Results"),
     title="🐟 Fish Species Classifier",
     description="Upload an image of a fish to identify its species",
-    examples=[
-        ["salmon.jpg"],
-        ["trout.jpg"]
-    ],
+    examples=valid_examples if valid_examples else None,  # Используем только существующие файлы
     allow_flagging="never"
 )
 
@@ -97,5 +112,6 @@ iface = gr.Interface(
 if __name__ == "__main__":
     iface.launch(
         server_name="0.0.0.0",
-        server_port=7860
+        server_port=7860,
+        cache_examples=False  # Отключаем кэширование примеров
     )