clean requirements.txt

.idea/workspace.xml

@@ -124,7 +124,7 @@
       <updated>1702301546100</updated>
       <workItem from="1702301554378" duration="4025000" />
       <workItem from="1702450051021" duration="10065000" />
-      <workItem from="1702476484538" duration="6510000" />
+      <workItem from="1702476484538" duration="7523000" />
     </task>
     <task id="LOCAL-00001" summary="init commit">
       <option name="closed" value="true" />
@@ -214,7 +214,15 @@
       <option name="project" value="LOCAL" />
       <updated>1702482463491</updated>
     </task>
-    <option name="localTasksCounter" value="12" />
+    <task id="LOCAL-00012" summary="clean requirements.txt">
+      <option name="closed" value="true" />
+      <created>1702483217316</created>
+      <option name="number" value="00012" />
+      <option name="presentableId" value="LOCAL-00012" />
+      <option name="project" value="LOCAL" />
+      <updated>1702483217316</updated>
+    </task>
+    <option name="localTasksCounter" value="13" />
     <servers />
   </component>
   <component name="TypeScriptGeneratedFilesManager">

Dockerfile

@@ -1,4 +1,5 @@
 FROM python:3.11-bookworm
+ARG HF_TOKEN
 RUN useradd -m -u 1000 user
 USER user
 # Set home to the user's home directory
@@ -11,5 +12,5 @@ RUN mkdir $HOME/.cache
 RUN chown -R user $HOME/.cache
 
 RUN pip install -r requirements.txt
-RUN python yolo_fire.py
+RUN HF_TOKEN=$HF_TOKEN python yolo_fire.py
 ENTRYPOINT ["streamlit", "run", "demo.py"]

requirements.txt

@@ -35,7 +35,7 @@ mdurl==0.1.2
 mpmath==1.3.0
 networkx==3.2.1
 numpy==1.26.2
-opencv-python==4.8.1.78
+opencv-contrib-python-headless==4.8.*
 packaging==23.2
 pandas==2.1.4
 Pillow==10.1.0
@@ -59,7 +59,6 @@ requests==2.31.0
 requests-toolbelt==1.0.0
 rich==13.7.0
 rpds-py==0.13.2
-sahi==0.11.15
 scipy==1.11.4
 seaborn==0.13.0
 shapely==2.0.2