Spaces:
Sleeping
Sleeping
Commit
·
74b41d6
1
Parent(s):
dd2e743
Update app.py
Browse files
app.py
CHANGED
@@ -42,7 +42,8 @@ if SAFETY_CHECKER == "True":
|
|
42 |
variant="fp16" if torch_dtype == torch.float16 else "fp32",
|
43 |
)
|
44 |
t2i_pipe = AutoPipelineForText2Image.from_pretrained(
|
45 |
-
"stabilityai/sdxl-turbo",
|
|
|
46 |
torch_dtype=torch_dtype,
|
47 |
variant="fp16" #if torch_dtype == torch.float16 else "fp32",
|
48 |
)
|
@@ -54,7 +55,8 @@ else:
|
|
54 |
variant="fp16" if torch_dtype == torch.float16 else "fp32",
|
55 |
)
|
56 |
t2i_pipe = AutoPipelineForText2Image.from_pretrained(
|
57 |
-
"stabilityai/sdxl-turbo",
|
|
|
58 |
safety_checker=None,
|
59 |
torch_dtype=torch_dtype,
|
60 |
variant="fp16" #if torch_dtype == torch.float16 else "fp32",
|
@@ -133,10 +135,8 @@ with gr.Blocks(css=css) as demo:
|
|
133 |
init_image_state = gr.State()
|
134 |
with gr.Column(elem_id="container"):
|
135 |
gr.Markdown(
|
136 |
-
"""#
|
137 |
-
##
|
138 |
-
SDXL Turbo model can generate high quality images in a single pass read more on [stability.ai post](https://stability.ai/news/stability-ai-sdxl-turbo).
|
139 |
-
**Model**: https://huggingface.co/stabilityai/sdxl-turbo
|
140 |
""",
|
141 |
elem_id="intro",
|
142 |
)
|
|
|
42 |
variant="fp16" if torch_dtype == torch.float16 else "fp32",
|
43 |
)
|
44 |
t2i_pipe = AutoPipelineForText2Image.from_pretrained(
|
45 |
+
#"stabilityai/sdxl-turbo",
|
46 |
+
"wanghuging/skin_demo",
|
47 |
torch_dtype=torch_dtype,
|
48 |
variant="fp16" #if torch_dtype == torch.float16 else "fp32",
|
49 |
)
|
|
|
55 |
variant="fp16" if torch_dtype == torch.float16 else "fp32",
|
56 |
)
|
57 |
t2i_pipe = AutoPipelineForText2Image.from_pretrained(
|
58 |
+
#"stabilityai/sdxl-turbo",
|
59 |
+
"wanghuging/skin_demo",
|
60 |
safety_checker=None,
|
61 |
torch_dtype=torch_dtype,
|
62 |
variant="fp16" #if torch_dtype == torch.float16 else "fp32",
|
|
|
135 |
init_image_state = gr.State()
|
136 |
with gr.Column(elem_id="container"):
|
137 |
gr.Markdown(
|
138 |
+
"""# Derm-T2IM Text to Image Skin Cancer
|
139 |
+
## GitHub:
|
|
|
|
|
140 |
""",
|
141 |
elem_id="intro",
|
142 |
)
|