basharat8763 commited on
Commit
ed4f8a5
·
verified ·
1 Parent(s): 6f6b160

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -2,8 +2,9 @@ from diffusers import DiffusionPipeline
2
  import torch
3
  import streamlit as st
4
 
5
- sdxl_base_model_path = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-xl-base-1.0")
6
- sdxl_refiner_model_path = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-xl-refiner-1.0")
 
7
 
8
  @st.cache_resource
9
  def load_pipeline():
 
2
  import torch
3
  import streamlit as st
4
 
5
+ # Model paths for Hugging Face
6
+ sdxl_base_model_path = "stabilityai/stable-diffusion-xl-base-1.0"
7
+ sdxl_refiner_model_path = "stabilityai/stable-diffusion-xl-refiner-1.0"
8
 
9
  @st.cache_resource
10
  def load_pipeline():