Spaces:
dreroc
/
Running on Zero

yichenchenchen commited on
Commit
0711bac
·
verified ·
1 Parent(s): 2114407

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -8
app.py CHANGED
@@ -6,12 +6,6 @@ import sys
6
  import time
7
  from inferencer import Inferencer
8
  from accelerate.utils import set_seed
9
-
10
- # sys.path.append("/mnt/data_vlm/chris/Harmon")
11
-
12
- # model_path = "/mnt/data_vlm/chris/Harmon/work_dirs/sft_stage4_mar_huge_lr_7e-6_edit_mainly_new_mar_few_steps_gpt_only_and_1024/500iters_hf/pytorch_model.bin"
13
- # ckpt_name = "/".join(model_path.split("/")[-3:])
14
- #model_path = "Skywork/Skywork-UniPic-1.5B"
15
  from huggingface_hub import snapshot_download
16
 
17
  model_dir = snapshot_download(repo_id="Skywork/Skywork-UniPic-1.5B")
@@ -207,7 +201,7 @@ with gr.Blocks(css=CSS) as demo:
207
  mode_state = gr.State(value="Understand Image")
208
 
209
  with gr.Tabs():
210
- with gr.Tab(f"Skywork Pic Chatbot (ckpt: {ckpt_name})", elem_id="tab_item_4"):
211
  chatbot = gr.Chatbot(
212
  elem_id="chatbot1",
213
  show_label=False,
@@ -219,7 +213,7 @@ with gr.Blocks(css=CSS) as demo:
219
  with gr.Row():
220
  mode_selector = gr.Radio(
221
  choices=["Generate Image","Edit Image","Understand Image"],
222
- value="Understand Image",
223
  label="Mode",
224
  interactive=True,
225
  )
 
6
  import time
7
  from inferencer import Inferencer
8
  from accelerate.utils import set_seed
 
 
 
 
 
 
9
  from huggingface_hub import snapshot_download
10
 
11
  model_dir = snapshot_download(repo_id="Skywork/Skywork-UniPic-1.5B")
 
201
  mode_state = gr.State(value="Understand Image")
202
 
203
  with gr.Tabs():
204
+ with gr.Tab("Skywork UniPic Chatbot", elem_id="tab_item_4"):
205
  chatbot = gr.Chatbot(
206
  elem_id="chatbot1",
207
  show_label=False,
 
213
  with gr.Row():
214
  mode_selector = gr.Radio(
215
  choices=["Generate Image","Edit Image","Understand Image"],
216
+ value="Generate Image",
217
  label="Mode",
218
  interactive=True,
219
  )