Saarthak2002 commited on
Commit
0461cbb
·
verified ·
1 Parent(s): 6c9d2f5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -7,7 +7,9 @@ import gradio as gr # Import Gradio for the interface
7
 
8
  #Function to generate image
9
  def generate_image(prompt, height=512, width=512):
10
- model_id = "stabilityai/stable-diffusion-2-1"
 
 
11
  HUGGINGFACE_API_KEY = os.getenv("HUGGINGFACE_API_KEY")
12
 
13
  if not HUGGINGFACE_API_KEY:
 
7
 
8
  #Function to generate image
9
  def generate_image(prompt, height=512, width=512):
10
+
11
+ model_id = "CompVis/stable-diffusion-v1-4"
12
+
13
  HUGGINGFACE_API_KEY = os.getenv("HUGGINGFACE_API_KEY")
14
 
15
  if not HUGGINGFACE_API_KEY: