Update app.py

app.py

@@ -35,6 +35,10 @@ if mps_available:
     torch_device = "cpu"
     torch_dtype = torch.float32
 
+# pipe = StableDiffusionPipeline.from_single_file(
+#     "https://huggingface.co/wanghuging/skin_demo/blob/main/skin_demo.safetensors"
+# )
+    
 if SAFETY_CHECKER == "True":
     i2i_pipe = AutoPipelineForImage2Image.from_pretrained(
         "stabilityai/sdxl-turbo",
@@ -43,7 +47,7 @@ if SAFETY_CHECKER == "True":
     )
     t2i_pipe = AutoPipelineForText2Image.from_pretrained(
         #"stabilityai/sdxl-turbo",
-        "wanghuging/skin_demo",
+        "wanghuging/demo_model",
         torch_dtype=torch_dtype,
         variant="fp16" #if torch_dtype == torch.float16 else "fp32",
     )
@@ -56,7 +60,7 @@ else:
     )
     t2i_pipe = AutoPipelineForText2Image.from_pretrained(
         #"stabilityai/sdxl-turbo",
-        "wanghuging/skin_demo",
+        "wanghuging/demo_model",
         safety_checker=None,
         torch_dtype=torch_dtype,
         variant="fp16" #if torch_dtype == torch.float16 else "fp32",
@@ -136,7 +140,8 @@ with gr.Blocks(css=css) as demo:
     with gr.Column(elem_id="container"):
         gr.Markdown(
             """# Derm-T2IM Text to Image Skin Cancer
-            ## GitHub: 
+            ## Demo
+            **Model**: https://huggingface.co/stabilityai/sdxl-turbo
             """,
             elem_id="intro",
         )