Vignesh455 commited on
Commit
70ff62b
1 Parent(s): a89c4c6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -17,7 +17,7 @@ def read_content(file_path: str) -> str:
17
 
18
  def predict(prompt, negative_prompt, guidance_scale, num_inference_steps,model, scheduler, lora, lora_weight):
19
  pipeline = diffusers.DiffusionPipeline.from_pretrained("SG161222/RealVisXL_V4.0", vae=vae).to("cuda")
20
- if model = 'Realistic_V6.0':
21
  pipeline = diffusers.DiffusionPipeline.from_pretrained("SG161222/Realistic_Vision_V6.0_B1_noVAE", vae=vae).to("cuda")
22
  pipeline.safety_checker = lambda images, **kwargs: (images, [False] * len(images))
23
  if model == "Realistic_V5.1":
 
17
 
18
  def predict(prompt, negative_prompt, guidance_scale, num_inference_steps,model, scheduler, lora, lora_weight):
19
  pipeline = diffusers.DiffusionPipeline.from_pretrained("SG161222/RealVisXL_V4.0", vae=vae).to("cuda")
20
+ if model == 'Realistic_V6.0':
21
  pipeline = diffusers.DiffusionPipeline.from_pretrained("SG161222/Realistic_Vision_V6.0_B1_noVAE", vae=vae).to("cuda")
22
  pipeline.safety_checker = lambda images, **kwargs: (images, [False] * len(images))
23
  if model == "Realistic_V5.1":