Hugging Face's logo

Spaces:
harmionestark
/
GraveImage
Runtime error

App Files Community
harmionestark commited on
Commit
620d2f0
Β·
verified Β·
1 Parent(s): 4ab97a1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -7,7 +7,7 @@ from PIL import Image
7
  app = Flask(__name__)
8
 
9
  # Load the model
10
- model_id = "kothariyashhh/GenAi-Texttoimage"
11
  pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
12
  pipe = pipe.to("cuda") # Ensure you're using a GPU for inference
13
 
 
7
  app = Flask(__name__)
8
 
9
  # Load the model
10
+ model_id = "sairajg/Text_To_Image"
11
  pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
12
  pipe = pipe.to("cuda") # Ensure you're using a GPU for inference
13