Hugging Face's logo

Spaces:
tranduy2610
/
test
Sleeping

App Files Community
tranduy2611 commited on
Commit
8184fca
·
1 Parent(s): 4bc8cc6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -15,7 +15,7 @@ app = FastAPI()
15
  hf_token = os.environ.get("HUGGINGFACE_TOKEN")
16
  login(token=hf_token)
17
  # Load the model
18
- model_id = "stabilityai/stable-diffusion-3.5-medium"
19
  device = "cuda" if torch.cuda.is_available() else "cpu"
20
  pipe = AutoPipelineForImage2Image.from_pretrained(
21
  model_id,
 
15
  hf_token = os.environ.get("HUGGINGFACE_TOKEN")
16
  login(token=hf_token)
17
  # Load the model
18
+ model_id = "kandinsky-community/kandinsky-2-2-decoder"
19
  device = "cuda" if torch.cuda.is_available() else "cpu"
20
  pipe = AutoPipelineForImage2Image.from_pretrained(
21
  model_id,