Hugging Face's logo

Spaces:
Lookingsoft-team
/
Text-Image-Diffusion
Runtime error

App Files Community
georgekutty george commited on
Commit
73dfd2c
·
1 Parent(s): 6ef0474

lkspace

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -11,7 +11,8 @@ from huggingface_hub import login
11
  login(token=os.getenv("HF_TOKEN"))
12
 
13
  # 4. Load private model using your token
14
- model_id = "Lookingsoft-team/Text_to_Image_Diffusion"
 
15
  device = "cuda" if torch.cuda.is_available() else "cpu"
16
 
17
  pipe = StableDiffusionPipeline.from_pretrained(
 
11
  login(token=os.getenv("HF_TOKEN"))
12
 
13
  # 4. Load private model using your token
14
+ model_id ="stabilityai/stable-diffusion-2-1-base"
15
+
16
  device = "cuda" if torch.cuda.is_available() else "cpu"
17
 
18
  pipe = StableDiffusionPipeline.from_pretrained(