TheVeshup commited on
Commit
c379683
·
verified ·
1 Parent(s): 743e9ca

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -12,7 +12,7 @@ if not HF_TOKEN:
12
  raise ValueError("Hugging Face token not found. Please set the 'HF_TOKEN' environment variable.")
13
 
14
  device = "cuda" if torch.cuda.is_available() else "cpu"
15
- model_repo_id = "stabilityai/stable-diffusion-3.5-large" # Replace with the model you would like to use
16
 
17
  if torch.cuda.is_available():
18
  torch_dtype = torch.float16
@@ -77,7 +77,7 @@ DEFAULT_GUIDANCE = 8
77
 
78
  with gr.Blocks(css=css) as demo:
79
  with gr.Column(elem_id="col-container"):
80
- gr.Markdown("<div id='header'><h1 id='title'>Veginator: Veshup's Image Generation AI</h1><p id='subtitle'>Create stunning images with just a prompt. Powered by cutting-edge AI technology.</p></div>")
81
 
82
  with gr.Row():
83
  prompt = gr.Text(
 
12
  raise ValueError("Hugging Face token not found. Please set the 'HF_TOKEN' environment variable.")
13
 
14
  device = "cuda" if torch.cuda.is_available() else "cpu"
15
+ model_repo_id = "CompVis/stable-diffusion-v1-4" # Replace with the model you would like to use
16
 
17
  if torch.cuda.is_available():
18
  torch_dtype = torch.float16
 
77
 
78
  with gr.Blocks(css=css) as demo:
79
  with gr.Column(elem_id="col-container"):
80
+ gr.Markdown("<div id='header'><h1 id='title'>Veshon: Veshup's Image Generation AI</h1><p id='subtitle'>Create stunning images with just a prompt. Powered by cutting-edge AI technology.</p></div>")
81
 
82
  with gr.Row():
83
  prompt = gr.Text(