andrewchwalik commited on
Commit
d4f70fe
·
verified ·
1 Parent(s): 0ee0afa

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -3
app.py CHANGED
@@ -8,10 +8,8 @@ token = os.getenv("HUGGINGFACE_TOKEN")
8
 
9
  print(f"Using Hugging Face Token: {os.getenv('HUGGINGFACE_TOKEN')}")
10
 
11
- # Authenticate with the Hugging Face token
12
  model = StableDiffusionPipeline.from_pretrained(
13
- "stabilityai/stable-diffusion-3.5-medium",
14
- use_auth_token=token
15
  )
16
  model.to("cuda") # Use GPU for faster processing (if available)
17
 
 
8
 
9
  print(f"Using Hugging Face Token: {os.getenv('HUGGINGFACE_TOKEN')}")
10
 
 
11
  model = StableDiffusionPipeline.from_pretrained(
12
+ "runwayml/stable-diffusion-v1-5"
 
13
  )
14
  model.to("cuda") # Use GPU for faster processing (if available)
15