Update app.py

app.py

@@ -34,8 +34,8 @@ def get_random_values(my_dict):
     return kind
 
 model_repo_id = "runwayml/stable-diffusion-v1-5"
-medsegfactory_id = "JohnWeck/StableDiffusion/checkpoint-300.pth"
-filename = 'checkpoint.pth'
+medsegfactory_id = "JohnWeck/StableDiffusion"
+filename = 'checkpoint-300.pth'
 
 device = "cuda" if torch.cuda.is_available() else "cpu"
 text_encoder = CLIPTextModel.from_pretrained(model_repo_id, subfolder="text_encoder")