update Dockerfile

Dockerfile

@@ -55,7 +55,8 @@ RUN pip install packaging && \
     pip install spacy && \
     pip install -U flash-attn gradio transformers fastapi tyro julius tiktoken hf-transfer && \
     pip install git+https://github.com/yuta0306/audiocraft.git && \
-    pip install -U --pre torch torch torchaudio --index-url https://download.pytorch.org/whl/nightly/cu121
+    pip install -U --pre torch torch torchaudio --index-url https://download.pytorch.org/whl/nightly/cu121 && \
+    pip install -U gradio
 ENV PYTHONPATH=${HOME}/app \
     PYTHONUNBUFFERED=1 \
     HF_HUB_ENABLE_HF_TRANSFER=1 \