DataScienceLover commited on
Commit
b106e83
·
verified ·
1 Parent(s): b09bb3a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -0
app.py CHANGED
@@ -2,6 +2,11 @@ import gradio as gr
2
  import cv2
3
  import numpy as np
4
  from ultralytics import YOLO
 
 
 
 
 
5
 
6
  # Загрузка модели (путь должен быть относительным или абсолютным)
7
  model = YOLO("models/best.pt")
 
2
  import cv2
3
  import numpy as np
4
  from ultralytics import YOLO
5
+ import os
6
+
7
+ os.environ["YOLO_CONFIG_DIR"] = "/tmp/ultralytics_config"
8
+ os.makedirs(os.environ["YOLO_CONFIG_DIR"], exist_ok=True)
9
+
10
 
11
  # Загрузка модели (путь должен быть относительным или абсолютным)
12
  model = YOLO("models/best.pt")