Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -22,8 +22,20 @@ CSS = """
|
|
| 22 |
.gradio-container {
|
| 23 |
max-width: 690px !important;
|
| 24 |
}
|
|
|
|
|
|
|
|
|
|
| 25 |
"""
|
| 26 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 27 |
# Ensure model and scheduler are initialized in GPU-enabled function
|
| 28 |
if torch.cuda.is_available():
|
| 29 |
unet = UNet2DConditionModel.from_config(base, subfolder="unet").to("cuda", torch.float16)
|
|
@@ -69,7 +81,7 @@ examples = [
|
|
| 69 |
|
| 70 |
# Gradio Interface
|
| 71 |
|
| 72 |
-
with gr.Blocks(css=CSS, theme="soft") as demo:
|
| 73 |
gr.HTML("<h1><center>Adobe DMD2🦖</center></h1>")
|
| 74 |
gr.HTML("<p><center><a href='https://huggingface.co/tianweiy/DMD2'>DMD2</a> text-to-image generation</center><br><center>Multi-Languages, 4-step is higher quality & 2X slower</center></p>")
|
| 75 |
with gr.Group():
|
|
|
|
| 22 |
.gradio-container {
|
| 23 |
max-width: 690px !important;
|
| 24 |
}
|
| 25 |
+
footer {
|
| 26 |
+
visibility: hidden;
|
| 27 |
+
}
|
| 28 |
"""
|
| 29 |
|
| 30 |
+
JS = """function () {
|
| 31 |
+
gradioURL = window.location.href
|
| 32 |
+
if (!gradioURL.endsWith('?__theme=dark')) {
|
| 33 |
+
window.location.replace(gradioURL + '?__theme=dark');
|
| 34 |
+
}
|
| 35 |
+
}"""
|
| 36 |
+
|
| 37 |
+
|
| 38 |
+
|
| 39 |
# Ensure model and scheduler are initialized in GPU-enabled function
|
| 40 |
if torch.cuda.is_available():
|
| 41 |
unet = UNet2DConditionModel.from_config(base, subfolder="unet").to("cuda", torch.float16)
|
|
|
|
| 81 |
|
| 82 |
# Gradio Interface
|
| 83 |
|
| 84 |
+
with gr.Blocks(css=CSS, js=JS, theme="soft") as demo:
|
| 85 |
gr.HTML("<h1><center>Adobe DMD2🦖</center></h1>")
|
| 86 |
gr.HTML("<p><center><a href='https://huggingface.co/tianweiy/DMD2'>DMD2</a> text-to-image generation</center><br><center>Multi-Languages, 4-step is higher quality & 2X slower</center></p>")
|
| 87 |
with gr.Group():
|