Update app.py

app.py

@@ -147,7 +147,7 @@ tokenizer, model, image_processor, context_len = load_pretrained_model(args.mode
 model = reload_from_ckpt("shi-labs/OLA-VLM-CLIP-ViT-Llama3-8b", model)
 our_chatbot = None
 
-pipe = StableUnCLIPImg2ImgPipeline.from_pretrained(f"stabilityai/stable-diffusion-2-1-unclip", torch_dtype=torch.float16, variant="fp16")
+pipe = StableUnCLIPImg2ImgPipeline.from_pretrained(f"radames/stable-diffusion-2-1-unclip-img2img", torch_dtype=torch.float16, variant="fp16")
 pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config)
 
 oneformer_processor = OneFormerProcessor.from_pretrained("shi-labs/oneformer_coco_swin_large")