Spaces:
Running
on
Zero
Running
on
Zero
oyly
commited on
Commit
·
2400b90
1
Parent(s):
14a1fe0
fix login
Browse files
app.py
CHANGED
@@ -11,8 +11,8 @@ import torch
|
|
11 |
import gradio as gr
|
12 |
import numpy as np
|
13 |
import spaces
|
14 |
-
|
15 |
-
|
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(
|
|
|
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()
|