Update app.py

app.py

@@ -6,15 +6,18 @@ import torch
 import random
 import uuid
 
-token = os.getenv("token")
-model = gr.load("models/Rojban/dreambooth4", hf_token=token)
+#token = os.getenv("token")
+#model = gr.load("models/Rojban/dreambooth4", hf_token=token)
+prj_path = "dreambooth4"
+
+model = "stabilityai/stable-diffusion-xl-base-1.0"
 
 pipe = DiffusionPipeline.from_pretrained(
     model,
     torch_dtype=torch.float16,
 )
 pipe.to("cuda")
-pipe.load_lora_weights("Rojban/dreambooth4", weight_name="pytorch_lora_weights.safetensors")
+pipe.load_lora_weights(prj_path, weight_name="pytorch_lora_weights.safetensors")
 
 refiner = StableDiffusionXLImg2ImgPipeline.from_pretrained(
     "stabilityai/stable-diffusion-xl-refiner-1.0",