docker-app

Dockerfile

@@ -0,0 +1,27 @@
+FROM python:3.9
+
+# Add settings taken from huggingface default container.
+WORKDIR /home/user/app
+RUN useradd -m -u 1000 user
+RUN apt-get update && apt-get install -y \
+    git git-lfs ffmpeg 	libsm6 	libxext6 \
+    cmake libgl1-mesa-glx \
+    && rm -rf /var/lib/apt/lists/* \
+    && git lfs install
+
+# Use older setuptools to avoid 2to3 error with dependencies.
+RUN pip install 'setuptools<58'
+
+# Install drexel_metadata dependencies
+COPY requirements.txt requirements.txt
+RUN pip install -r requirements.txt
+
+# Install gradio after main dependencies to prevent
+# `cannot import name 'deprecated' from 'typing_extensions'` error.
+RUN pip install gradio
+
+# Cache matplotlib fonts to improve startup time.
+RUN python -c "import matplotlib.pyplot"
+
+COPY . /home/user/app
+CMD python app.py

README.md

@@ -1,13 +1,11 @@
 ---
 title: Drexel Metadata App
-emoji: 🐢
+emoji: 🐟
 colorFrom: gray
 colorTo: red
-sdk: gradio
-sdk_version: 3.6
+sdk: docker
 app_file: app.py
 pinned: false
-python_version: 3.9.18
 license: mit
 ---
 

app.py

@@ -73,5 +73,5 @@ dm_app = gr.Interface(
      allow_flagging="never", # Do not save user's results or prompt for users to save the results
      examples=EXAMPLES,
 )
-dm_app.launch()
+dm_app.launch(server_name="0.0.0.0")
 

app_header.md

@@ -1,3 +1,4 @@
 # Drexel Metadata
-Generate Metadata for a fish image using the [hdr-bgnn/drexel_metadata model](https://github.com/hdr-bgnn/drexel_metadata).
+Generate object detection metadata for a fish image using the [imageomics/Drexel-metadata-generator model](https://huggingface.co/imageomics/Drexel-metadata-generator).
 The model will create a visualization based on the image, a mask of the fish outline, and JSON metadata.
+The source code for the model is available at [hdr-bgnn/drexel_metadata model](https://github.com/hdr-bgnn/drexel_metadata).