Update app.py

app.py

@@ -32,8 +32,8 @@ DEFAULT_WIDTH = 1024
 
 headers = {"Content-Type": "application/json", "Authorization": f"Bearer {HF_TOKEN}" }
 
-repo_large_path = snapshot_download(repo_id="stabilityai/stable-diffusion-3-medium", revision="refs/pr/26", repo_type="model", token=HF_TOKEN)
-repo_large_transformer_path = SD3Transformer2DModel.from_pretrained(repo_large_path, torch_dtype=torch.float16)
+repo_large_path = snapshot_download(repo_id="stabilityai/stable-diffusion-3-medium", revision="refs/pr/26", token=HF_TOKEN)
+repo_large_transformer_path = SD3Transformer2DModel.from_pretrained(repo_large_path, subfolder="transformer", torch_dtype=torch.float16)
 repo_large_transformer = PeftModel.from_pretrained(repo_large_transformer_path, "jasperai/flash-sd3")
 
 css = '''