Upload 3 files

app.py

@@ -0,0 +1,59 @@
+import gradio as gr
+from huggingface_hub import hf_hub_download
+from ultralytics import YOLO
+from PIL import Image
+
+# Скачиваем веса из вашего репо на Hugging Face
+REPO_ID = "shaquilledanil/yolo11-skinai"
+FILENAME = "best.pt"
+WEIGHT_PATH = hf_hub_download(repo_id=REPO_ID, filename=FILENAME)
+
+# Загружаем YOLO-модель
+model = YOLO(WEIGHT_PATH)
+
+def yolo_inference(image: Image.Image):
+    """
+    Принимает PIL.Image, прогоняет через YOLO-модель, 
+    возвращает кортеж (annotated_image, detections_info).
+    """
+    results = model.predict(source=image)
+
+    # Подготовим изображение с нарисованными bboxes
+    # Ultralytics в results[0].plot() вернёт PIL.Image с рамками
+    annotated_image = results[0].plot() if len(results) else image
+
+    # Соберём текстовую информацию о детекциях
+    detections = []
+    if len(results):
+        boxes = results[0].boxes
+        for box in boxes:
+            cls_id = int(box.cls[0].item())
+            conf = float(box.conf[0].item())
+            xyxy = box.xyxy[0].tolist()
+            class_name = results[0].names[cls_id]
+            detections.append({
+                "class_id": cls_id,
+                "class_name": class_name,
+                "confidence": round(conf, 3),
+                "bbox": [round(x, 1) for x in xyxy]
+            })
+
+    return annotated_image, detections
+
+# Создаём Gradio-интерфейс:
+#  - inputs: одна картинка
+#  - outputs: (1) картинка с боксами, (2) JSON (список детекций)
+demo = gr.Interface(
+    fn=yolo_inference,
+    inputs=gr.Image(type="pil"),
+    outputs=[
+        gr.Image(type="pil", label="Detected Objects"), 
+        "json"
+    ],
+    title="YOLO SkinAI Demo",
+    description="Загрузите изображение, чтобы увидеть результат детекции"
+)
+
+# Запуск (при локальном тесте):
+if __name__ == "__main__":
+    demo.launch()

best.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:9fe1180af4d10f963eb6f41fa1f5df8eff4e2a5c9be1d03277fdc968b8716aec
+size 3212411

requirements.txt

@@ -0,0 +1,151 @@
+aiofiles==23.2.1
+aiohttp==3.9.5
+aiohttp-socks==0.8.4
+aiosignal==1.3.1
+annotated-types==0.7.0
+anyio==4.8.0
+asttokens==3.0.0
+async-timeout==4.0.3
+attrs==24.2.0
+blinker==1.8.2
+build==1.2.2.post1
+CacheControl==0.14.0
+certifi==2024.8.30
+charset-normalizer==3.4.0
+cleo==2.1.0
+click==8.1.7
+colorama==0.4.6
+comm==0.2.2
+contourpy==1.3.1
+crashtest==0.4.1
+cycler==0.12.1
+Cython==3.0.11
+debugpy==1.8.11
+decorator==5.1.1
+distlib==0.3.9
+dtw-python==1.5.3
+dulwich==0.21.7
+executing==2.1.0
+fake-useragent==1.5.1
+Faker==27.0.0
+fastapi==0.115.6
+fastjsonschema==2.20.0
+ffmpy==0.5.0
+filelock==3.16.1
+Flask==3.0.3
+fonttools==4.55.3
+frozenlist==1.4.1
+fsspec==2024.10.0
+gradio==5.10.0
+gradio_client==1.5.3
+h11==0.14.0
+httpcore==1.0.7
+httpx==0.28.1
+huggingface-hub==0.27.1
+idna==3.8
+installer==0.7.0
+ipykernel==6.29.5
+ipython==8.30.0
+itsdangerous==2.2.0
+jaraco.classes==3.4.0
+jedi==0.19.2
+Jinja2==3.1.4
+jupyter_client==8.6.3
+jupyter_core==5.7.2
+keyring==24.3.1
+kiwisolver==1.4.7
+llvmlite==0.43.0
+loguru==0.7.2
+markdown-it-py==3.0.0
+MarkupSafe==2.1.5
+matplotlib==3.10.0
+matplotlib-inline==0.1.7
+mdurl==0.1.2
+more-itertools==10.5.0
+mpmath==1.3.0
+msgpack==1.1.0
+multidict==6.0.5
+nest-asyncio==1.6.0
+networkx==3.4.2
+numba==0.60.0
+numpy==2.0.2
+openai-whisper==20240930
+opencv-python==4.10.0.84
+orjson==3.10.13
+packaging==24.1
+pandas==2.2.2
+parso==0.8.4
+pexpect==4.9.0
+Pillow==9.5.0
+pkginfo==1.11.2
+platformdirs==4.3.6
+poetry==1.8.4
+poetry-core==1.9.1
+poetry-plugin-export==1.8.0
+prompt_toolkit==3.0.48
+psutil==6.1.0
+ptyprocess==0.7.0
+pure_eval==0.2.3
+py-cpuinfo==9.0.0
+pyaes==1.6.1
+pydantic==2.10.4
+pydantic_core==2.27.2
+pydub==0.25.1
+Pygments==2.18.0
+pyparsing==3.2.0
+pyproject_hooks==1.2.0
+Pyrogram==2.0.106
+PySocks==1.7.1
+python-dateutil==2.9.0.post0
+python-multipart==0.0.20
+python-socks==2.5.1
+pytz==2024.1
+pywin32==308
+pywin32-ctypes==0.2.3
+PyYAML==6.0.2
+pyzmq==26.2.0
+RapidFuzz==3.10.0
+regex==2024.9.11
+requests==2.32.3
+requests-toolbelt==1.0.0
+rich==13.9.4
+ruff==0.8.6
+safehttpx==0.1.6
+safetensors==0.4.5
+scipy==1.14.1
+seaborn==0.13.2
+semantic-version==2.10.0
+sentry-sdk==2.19.2
+setuptools==75.3.0
+shellingham==1.5.4
+six==1.16.0
+sniffio==1.3.1
+stack-data==0.6.3
+starlette==0.41.3
+sympy==1.13.1
+telegram==0.0.1
+TgCrypto==1.2.5
+thop==0.1.1.post2209072238
+tiktoken==0.8.0
+tokenizers==0.20.3
+tomlkit==0.13.2
+torch==2.5.1
+torchvision==0.20.1
+tornado==6.4.2
+tqdm==4.66.6
+traitlets==5.14.3
+transformers==4.46.2
+trove-classifiers==2024.10.16
+typer==0.15.1
+typing_extensions==4.12.2
+tzdata==2024.1
+ultralytics==8.3.57
+ultralytics-thop==2.0.13
+urllib3==2.2.3
+uvicorn==0.22.0
+virtualenv==20.27.0
+wcwidth==0.2.13
+websockets==14.1
+Werkzeug==3.1.1
+win32-setctime==1.1.0
+yarl==1.10.0