Update app.py

app.py

@@ -6,7 +6,7 @@ from diffusers import StableDiffusionPipeline, DDIMScheduler
 from IPython.display import display
 from text_generation import Client, InferenceAPIClient
 
-model_path = WEIGHTS_DIR             # If you want to use previously trained model saved in gdrive, replace this with the full path of model in gdrive
+model_path = "shoe.ckpt"
 
 pipe = StableDiffusionPipeline.from_pretrained(model_path, safety_checker=None, torch_dtype=torch.float16).to("cuda")
 pipe.scheduler = DDIMScheduler.from_config(pipe.scheduler.config)