Rooni commited on
Commit
30128cf
·
verified ·
1 Parent(s): f240016

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -8
app.py CHANGED
@@ -16,9 +16,9 @@ from diffusers import StableDiffusionXLPipeline, StableDiffusionXLImg2ImgPipelin
16
  logging.basicConfig(level=logging.INFO)
17
  logger = logging.getLogger(__name__)
18
 
19
- DESCRIPTION = "Animagine XL 3.1"
20
- if not torch.cuda.is_available():
21
- DESCRIPTION += "\n<p>Running on CPU 🥶 This demo does not work on CPU. </p>"
22
  IS_COLAB = utils.is_google_colab() or os.getenv("IS_COLAB") == "1"
23
  HF_TOKEN = os.getenv("HF_TOKEN")
24
  CACHE_EXAMPLES = torch.cuda.is_available() and os.getenv("CACHE_EXAMPLES") == "1"
@@ -206,11 +206,11 @@ quality_prompt = {
206
 
207
  wildcard_files = utils.load_wildcard_files("wildcard")
208
 
209
- with gr.Blocks(css="style.css", theme="NoCrypt/[email protected]") as demo:
210
- title = gr.HTML(
211
- f"""<h1><span>{DESCRIPTION}</span></h1>""",
212
- elem_id="title",
213
- )
214
  gr.Markdown(
215
  f"""Gradio demo for [cagliostrolab/animagine-xl-3.1](https://huggingface.co/cagliostrolab/animagine-xl-3.1)""",
216
  elem_id="subtitle",
 
16
  logging.basicConfig(level=logging.INFO)
17
  logger = logging.getLogger(__name__)
18
 
19
+ #DESCRIPTION = "Animagine XL 3.1"
20
+ #if not torch.cuda.is_available():
21
+ # DESCRIPTION += "\n<p>Running on CPU 🥶 This demo does not work on CPU. </p>"
22
  IS_COLAB = utils.is_google_colab() or os.getenv("IS_COLAB") == "1"
23
  HF_TOKEN = os.getenv("HF_TOKEN")
24
  CACHE_EXAMPLES = torch.cuda.is_available() and os.getenv("CACHE_EXAMPLES") == "1"
 
206
 
207
  wildcard_files = utils.load_wildcard_files("wildcard")
208
 
209
+ with gr.Blocks(css="style.css") as demo:
210
+ # title = gr.HTML(
211
+ # f"""<h1><span>{DESCRIPTION}</span></h1>""",
212
+ # elem_id="title",
213
+ # )
214
  gr.Markdown(
215
  f"""Gradio demo for [cagliostrolab/animagine-xl-3.1](https://huggingface.co/cagliostrolab/animagine-xl-3.1)""",
216
  elem_id="subtitle",