wanghuging commited on
Commit
74b41d6
·
1 Parent(s): dd2e743

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -6
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
- """# SDXL Turbo Image to Image/Text to Image
137
- ## Unofficial Demo
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
  )