Update Dockerfile

Dockerfile

@@ -26,27 +26,20 @@ RUN git clone --branch main https://github.com/ack00gar/FunGen-AI-Powered-Funscr
     git config user.email "bot@example.com" && \
     git config user.name "Bot"
 
-# Patch FunGen to disable dependency check
-RUN cd FunGen && \
-    echo "def check_application_dependencies():" > core/application_dependencies_patch.py && \
-    echo "    pass" >> core/application_dependencies_patch.py && \
-    sed -i '1s/^/from core.application_dependencies_patch import check_application_dependencies as _noop\nimport core.application_dependencies\ncore.application_dependencies.check_application_dependencies = _noop\n/' main.py
+WORKDIR /FunGen
 
-RUN pip install -r FunGen/cuda.requirements.txt && \
-    pip install tensorrt
+RUN pip install -r cuda.requirements.txt
 
-RUN pip install gradio==4.26.0 gradio-client==0.15.1
+RUN pip install tensorrt torch-tensorrt --extra-index-url https://download.pytorch.org/whl/nightly/cu128 || true
 
-RUN mkdir -p /tmp/uploads /tmp/outputs /app/models
-
-WORKDIR /app/FunGen
-
-RUN pip install -r core.requirements.txt
+RUN pip install gradio==4.16.0 huggingface-hub==0.19.3 pydantic>=2.0
 
-RUN python -m pip install --upgrade pip setuptools wheel
+RUN mkdir -p /tmp/uploads /tmp/outputs /app/models
 
 WORKDIR /app
 
 COPY app.py /app/app.py
 
+ENV FUNGEN_SKIP_CHECK=1
+
 CMD ["python", "/app/app.py"]