dependencies

Dockerfile

@@ -11,6 +11,7 @@ RUN apt-get update && apt-get install -y \
     ffmpeg \
     libsm6 \
     libxext6 \
+    curl \
     && rm -rf /var/lib/apt/lists/*
 
 # Ensure we're using Python 3.10
@@ -25,18 +26,32 @@ RUN git clone https://github.com/hkchengrex/MMAudio.git
 # Set working directory to MMAudio
 WORKDIR /code/MMAudio
 
-# Install dependencies
-RUN pip3 install --no-cache-dir numpy && \
-    pip3 install --no-cache-dir torch==2.1.2 torchvision==0.16.2 torchaudio==2.1.2 --index-url https://download.pytorch.org/whl/cu118 && \
-    pip3 install --no-cache-dir colorlog && \
-    pip3 install --no-cache-dir -r requirements.txt && \
-    pip3 install -e .
+# Install base dependencies first
+RUN pip3 install --no-cache-dir \
+    setuptools \
+    wheel \
+    numpy==1.24.3 \
+    colorlog \
+    typing_extensions
+
+# Install PyTorch and related packages
+RUN pip3 install --no-cache-dir \
+    torch==2.1.2 \
+    torchvision==0.16.2 \
+    torchaudio==2.1.2 \
+    --index-url https://download.pytorch.org/whl/cu118
+
+# Install other dependencies
+RUN pip3 install --no-cache-dir -r requirements.txt
+
+# Install MMAudio
+RUN pip3 install -e .
 
 # Create output directory
 RUN mkdir -p output/gradio && chmod 777 output/gradio
 
-# Rename gradio_demo.py to app.py
-RUN mv gradio_demo.py app.py
+# Copy app.py (we'll use our own version instead of the one from the repo)
+COPY app.py .
 
 # Set environment variables for Hugging Face Spaces
 ENV PYTHONUNBUFFERED=1

app.py

@@ -232,5 +232,8 @@ if __name__ == "__main__":
     args = parser.parse_args()
 
     gr.TabbedInterface([video_to_audio_tab, text_to_audio_tab, image_to_audio_tab],
-                       ['Video-to-Audio', 'Text-to-Audio', 'Image-to-Audio (experimental)']).launch(auth=("admin", "sonisphere"),share=True,
-                           server_port=args.port, allowed_paths=[output_dir])
+                       ['Video-to-Audio', 'Text-to-Audio', 'Image-to-Audio (experimental)']).launch(
+                           server_name="0.0.0.0",
+                           server_port=7860,
+                           auth=("admin", "sonisphere"),
+                           allowed_paths=[output_dir])