amildravid4292 commited on
Commit
8ba180b
·
verified ·
1 Parent(s): e01440c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -0
app.py CHANGED
@@ -387,6 +387,10 @@ intro = """
387
 
388
 
389
  with gr.Blocks(css="style.css") as demo:
 
 
 
 
390
  gr.HTML(intro)
391
 
392
  gr.Markdown("""<div style="text-align: justify;"> In this demo, you can get an identity-encoding model by sampling or inverting. To use a model previously downloaded from this demo see \"Uploading a model\" in the Advanced Options. Next, you can generate new images from it, or edit the identity encoded in the model and generate images from the edited model. We provide detailed instructions and tips at the bottom of the page.""")
 
387
 
388
 
389
  with gr.Blocks(css="style.css") as demo:
390
+
391
+ unet, vae, text_encoder, tokenizer, noise_scheduler = load_models(device)
392
+ network = None
393
+
394
  gr.HTML(intro)
395
 
396
  gr.Markdown("""<div style="text-align: justify;"> In this demo, you can get an identity-encoding model by sampling or inverting. To use a model previously downloaded from this demo see \"Uploading a model\" in the Advanced Options. Next, you can generate new images from it, or edit the identity encoded in the model and generate images from the edited model. We provide detailed instructions and tips at the bottom of the page.""")