Spaces:

Lookingsoft-team
/

Text-Image-Diffusion

Runtime error

Lookingsoft commited on Jun 1

Commit

8eabfed

verified ·

1 Parent(s): 257bf5a

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -7,7 +7,7 @@ from spaces import GPU  # 👈 only for HF Spaces
 login(token=os.getenv("HF_TOKEN"))
-model_id = "runwayml/stable-diffusion-v1-5"
 device = "cuda" if torch.cuda.is_available() else "cpu"
 pipe = StableDiffusionPipeline.from_pretrained(

 login(token=os.getenv("HF_TOKEN"))
+model_id = "Lookingsoft-team/Text_to_Image_Diffusion"
 device = "cuda" if torch.cuda.is_available() else "cpu"
 pipe = StableDiffusionPipeline.from_pretrained(