Spaces:
Running
on
Zero
Running
on
Zero
iimmortall
commited on
Commit
•
df5a6e4
1
Parent(s):
92229a9
update
Browse files
app.py
CHANGED
@@ -14,7 +14,6 @@ from huggingface_hub import hf_hub_download, snapshot_download
|
|
14 |
|
15 |
model_name = "iimmortall/UltraFusion"
|
16 |
auth_token = os.getenv("HF_AUTH_TOKEN")
|
17 |
-
# print(auth_token)
|
18 |
# greet_file = hf_hub_download(repo_id=model_name, filename="main.py", use_auth_token=auth_token)
|
19 |
# sys.path.append(os.path.split(greet_file)[0])
|
20 |
model_folder = snapshot_download(repo_id=model_name, token=auth_token, local_dir="/home/user/app")
|
@@ -142,10 +141,12 @@ with gr.Blocks(css=css) as demo:
|
|
142 |
|
143 |
gr.Examples(
|
144 |
examples=examples,
|
145 |
-
inputs=[under_expo_img, over_expo_img],
|
146 |
label="Examples",
|
147 |
# examples_per_page=10,
|
|
|
148 |
cache_examples=True,
|
|
|
149 |
)
|
150 |
# gr.Markdown(_CITE_)
|
151 |
run_button.click(fn=infer,
|
|
|
14 |
|
15 |
model_name = "iimmortall/UltraFusion"
|
16 |
auth_token = os.getenv("HF_AUTH_TOKEN")
|
|
|
17 |
# greet_file = hf_hub_download(repo_id=model_name, filename="main.py", use_auth_token=auth_token)
|
18 |
# sys.path.append(os.path.split(greet_file)[0])
|
19 |
model_folder = snapshot_download(repo_id=model_name, token=auth_token, local_dir="/home/user/app")
|
|
|
141 |
|
142 |
gr.Examples(
|
143 |
examples=examples,
|
144 |
+
inputs=[under_expo_img, over_expo_img, num_inference_steps],
|
145 |
label="Examples",
|
146 |
# examples_per_page=10,
|
147 |
+
fn=infer,
|
148 |
cache_examples=True,
|
149 |
+
outputs=[result,],
|
150 |
)
|
151 |
# gr.Markdown(_CITE_)
|
152 |
run_button.click(fn=infer,
|