oyly commited on
Commit
2400b90
·
1 Parent(s): 14a1fe0
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -11,8 +11,8 @@ import torch
11
  import gradio as gr
12
  import numpy as np
13
  import spaces
14
- # from huggingface_hub import login
15
- # login(token=os.getenv('Token'))
16
  from flux.sampling_lore import denoise, get_schedule, prepare, unpack, get_v_mask, add_masked_noise_to_z,get_mask_one_tensor, denoise_with_noise_optim,prepare_tokens
17
  from flux.util_lore import (configs, embed_watermark, load_ae, load_clip,
18
  load_flow_model, load_t5)
@@ -350,4 +350,4 @@ def create_demo(model_name: str):
350
  return demo
351
 
352
  demo = create_demo("flux-dev")
353
- demo.launch(share=True)
 
11
  import gradio as gr
12
  import numpy as np
13
  import spaces
14
+ from huggingface_hub import login
15
+ login(token=os.getenv('Token'))
16
  from flux.sampling_lore import denoise, get_schedule, prepare, unpack, get_v_mask, add_masked_noise_to_z,get_mask_one_tensor, denoise_with_noise_optim,prepare_tokens
17
  from flux.util_lore import (configs, embed_watermark, load_ae, load_clip,
18
  load_flow_model, load_t5)
 
350
  return demo
351
 
352
  demo = create_demo("flux-dev")
353
+ demo.launch()