Update app.py

app.py

@@ -9,7 +9,7 @@ def run(checkpoint, prompt):
     model_id = checkpoint
     adapter_id = "latent-consistency/lcm-lora-sdv1-5"
 
-    pipe = AutoPipelineForText2Image.from_single_file(model_id, torch_dtype=torch.float32, safety_checker=None).to("cpu")
+    pipe = AutoPipelineForText2Image.from_pretrained(model_id, torch_dtype=torch.float32, safety_checker=None).to("cpu")
     pipe.scheduler = LCMScheduler.from_config(pipe.scheduler.config)
 
     # load and fuse lcm lora
@@ -21,7 +21,7 @@ def run(checkpoint, prompt):
     return image
 
 with gr.Blocks() as demo:
-    input_checkpoint = gr.Text(value="https://huggingface.co/gsdf/Counterfeit-V3.0/blob/main/Counterfeit-V3.0_fp16.safetensors", label="Checkpoint")
+    input_checkpoint = gr.Text(value="Lykon/dreamshaper-8", label="Checkpoint")
     input_prompt = gr.Text(value="Self-portrait oil painting, a beautiful cyborg with golden hair, 8k", label="Prompt")
     out = gr.Image(type="pil")
     btn = gr.Button("Run")