prithivMLmods commited on
Commit
0cc6029
1 Parent(s): 37a94aa

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -247,7 +247,7 @@ device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
247
  #Compile
248
  if torch.cuda.is_available():
249
  pipe = StableDiffusionXLPipeline.from_pretrained(
250
- "SG161222/RealVisXL_V4.0_Lightning",
251
  torch_dtype=torch.float16,
252
  use_safetensors=True,
253
  add_watermarker=False,
@@ -532,8 +532,8 @@ with gr.Blocks(css=css, theme="bethecloud/storj_theme") as demo:
532
  api_name="run",
533
  )
534
  # Adding a predefined gallery
535
- gr.Markdown("### Generated Images")
536
- predefined_gallery = gr.Gallery(label="Generated Images", columns=3, show_label=False, value=load_predefined_images())
537
  #gr.Markdown("⚠️ responsible for ensuring it meets appropriate ethical standards")
538
  gr.Markdown(DESCRIPTIONy)
539
 
 
247
  #Compile
248
  if torch.cuda.is_available():
249
  pipe = StableDiffusionXLPipeline.from_pretrained(
250
+ "SG161222/RealVisXL_V4.0",
251
  torch_dtype=torch.float16,
252
  use_safetensors=True,
253
  add_watermarker=False,
 
532
  api_name="run",
533
  )
534
  # Adding a predefined gallery
535
+ gr.Markdown("### Gallery Images")
536
+ predefined_gallery = gr.Gallery(label="Gallery Images", columns=3, show_label=False, value=load_predefined_images())
537
  #gr.Markdown("⚠️ responsible for ensuring it meets appropriate ethical standards")
538
  gr.Markdown(DESCRIPTIONy)
539