Update app.py

app.py

@@ -4,10 +4,11 @@ import random
 
 # import spaces #[uncomment to use ZeroGPU]
 from diffusers import DiffusionPipeline
+from diffusers import StableDiffusionPipeline
 import torch
 
 device = "cuda" if torch.cuda.is_available() else "cpu"
-model_repo_id = "table-diffusion-v1-5/stable-diffusion-v1-5"  # Replace to the model you would like to use
+model_repo_id = "stable-diffusion-v1-5/stable-diffusion-v1-5"  # Replace to the model you would like to use
 
 if torch.cuda.is_available():
     torch_dtype = torch.float16