sd-vae-ft-ms

Dockerfile

@@ -3,9 +3,12 @@ FROM artifacts.iflytek.com/docker-private/atp/base_image_for_ailab:0.0.1
 
 WORKDIR /app
 
-COPY . /app
+COPY requirements.txt /app
 
 RUN pip config set global.index-url https://pypi.mirrors.ustc.edu.cn/simple 
 RUN pip install -r requirements.txt
 
+COPY . /app
+
+
 CMD ["python", "app.py"]

app.py

@@ -25,8 +25,8 @@ def text2image(prompt):
 demo = gr.Interface(fn=text2image,
                     inputs='text',
                     outputs='image',
-                    title = "text2image",
                     theme = theme,
+                    css = "footer {visibility: hidden}",
                     examples = ['a photo of an astronaut riding a horse on mars'])