Update

Dockerfile

@@ -1,53 +0,0 @@
-FROM nvidia/cuda:11.7.1-cudnn8-devel-ubuntu22.04
-ENV DEBIAN_FRONTEND=noninteractive
-RUN apt-get update && \
-    apt-get upgrade -y && \
-    apt-get install -y --no-install-recommends \
-    git \
-    git-lfs \
-    wget \
-    curl \
-    # python build dependencies \
-    build-essential \
-    libssl-dev \
-    zlib1g-dev \
-    libbz2-dev \
-    libreadline-dev \
-    libsqlite3-dev \
-    libncursesw5-dev \
-    xz-utils \
-    tk-dev \
-    libxml2-dev \
-    libxmlsec1-dev \
-    libffi-dev \
-    liblzma-dev && \
-    apt-get clean && \
-    rm -rf /var/lib/apt/lists/*
-
-RUN useradd -m -u 1000 user
-USER user
-ENV HOME=/home/user \
-    PATH=/home/user/.local/bin:${PATH}
-WORKDIR ${HOME}/app
-
-RUN curl https://pyenv.run | bash
-ENV PATH=${HOME}/.pyenv/shims:${HOME}/.pyenv/bin:${PATH}
-ARG PYTHON_VERSION=3.10.11
-RUN pyenv install ${PYTHON_VERSION} && \
-    pyenv global ${PYTHON_VERSION} && \
-    pyenv rehash && \
-    pip install --no-cache-dir -U pip setuptools wheel
-
-RUN pip install --no-cache-dir -U torch==2.0.1 torchvision==0.15.2
-COPY --chown=1000 requirements.txt /tmp/requirements.txt
-RUN pip install --no-cache-dir -U -r /tmp/requirements.txt
-
-COPY --chown=1000 . ${HOME}/app
-ENV PYTHONPATH=${HOME}/app \
-    PYTHONUNBUFFERED=1 \
-    GRADIO_ALLOW_FLAGGING=never \
-    GRADIO_NUM_PORTS=1 \
-    GRADIO_SERVER_NAME=0.0.0.0 \
-    GRADIO_THEME=huggingface \
-    SYSTEM=spaces
-CMD ["python", "app.py"]

README.md

@@ -3,7 +3,8 @@ title: UniDiffuser
 emoji: 😻
 colorFrom: gray
 colorTo: green
-sdk: docker
+sdk: gradio
+skk_version: 3.46.0
 app_file: app.py
 pinned: false
 license: other

app.py

@@ -8,6 +8,7 @@ import random
 import gradio as gr
 import numpy as np
 import PIL.Image
+import spaces
 import torch
 from diffusers import UniDiffuserPipeline
 
@@ -32,6 +33,7 @@ if torch.cuda.is_available():
     pipe.to(device)
 
 
+@spaces.GPU
 def run(
     mode: str,
     prompt: str,

requirements.txt

@@ -1,9 +1,7 @@
 accelerate==0.23.0
 diffusers==0.21.4
 gradio==3.46.0
-huggingface-hub==0.17.3
-torch==2.0.1
-torchvision==0.15.2
+spaces==0.15.0
+torch==2.0.0
+torchvision==0.15.1
 transformers==4.34.0
-triton==2.0.0
-xformers==0.0.20