Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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 = "
|
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'>
|
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(
|