<feat> ZeroGPU only supports gradio sdk.

.gitattributes

@@ -32,5 +32,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.xz filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
-*tfevents* filter=lfs diff=lfs merge=lfs -text
-*.py text eol=lf
+*tfevents* filter=lfs diff=lfs merge=lfs -text

Dockerfile

@@ -1,17 +0,0 @@
-FROM pytorch/pytorch:2.5.1-cuda12.1-cudnn8-runtime
-
-RUN apt-get update && apt-get install -y \
-    git build-essential python3-dev ninja-build && \
-    rm -rf /var/lib/apt/lists/*
-
-RUN pip install --upgrade pip
-
-COPY requirements.txt /tmp/requirements.txt
-RUN pip install -r /tmp/requirements.txt
-
-RUN pip install git+https://github.com/Dao-AILab/flash-attention.git@v2.7.3#subdirectory=csrc
-
-COPY . /app
-WORKDIR /app
-
-CMD ["python", "app.py"]

README.md

@@ -3,7 +3,9 @@ title: DRA-Ctrl
 emoji: 😻
 colorFrom: green
 colorTo: gray
-sdk: docker
+sdk: gradio
+sdk_version: 5.33.2
+app_file: app.py
 pinned: false
 license: apache-2.0
 short_description: huggingface space for DRA-Ctrl.