iimmortall commited on
Commit
df5a6e4
1 Parent(s): 92229a9
Files changed (1) hide show
  1. app.py +3 -2
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,