howard-hou commited on
Commit
922f821
1 Parent(s): fbe0b0a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -8,7 +8,6 @@ from transformers import CLIPImageProcessor
8
  from huggingface_hub import hf_hub_download
9
 
10
  ctx_limit = 3500
11
- num_image_embeddings = 4096
12
  title = 'ViusualRWKV-v5'
13
  rwkv_remote_path = "rwkv1b5-vitl336p14-577token_mix665k_rwkv.pth"
14
  vision_remote_path = "rwkv1b5-vitl336p14-577token_mix665k_visual.pth"
@@ -143,13 +142,13 @@ with gr.Blocks(title=title) as demo:
143
  with gr.Column():
144
  image = gr.Image(type='pil', label="Image")
145
  with gr.Column():
146
- prompt = gr.Textbox(lines=5, label="Prompt",
147
  value="Render a clear and concise summary of the photo.")
148
  with gr.Row():
149
  submit = gr.Button("Submit", variant="primary")
150
  clear = gr.Button("Clear", variant="secondary")
151
  with gr.Column():
152
- output = gr.Textbox(label="Output", lines=7)
153
  data = gr.Dataset(components=[image, prompt], samples=examples, label="Examples", headers=["Image", "Prompt"])
154
  submit.click(chatbot, [image, prompt], [output])
155
  clear.click(lambda: None, [], [output])
 
8
  from huggingface_hub import hf_hub_download
9
 
10
  ctx_limit = 3500
 
11
  title = 'ViusualRWKV-v5'
12
  rwkv_remote_path = "rwkv1b5-vitl336p14-577token_mix665k_rwkv.pth"
13
  vision_remote_path = "rwkv1b5-vitl336p14-577token_mix665k_visual.pth"
 
142
  with gr.Column():
143
  image = gr.Image(type='pil', label="Image")
144
  with gr.Column():
145
+ prompt = gr.Textbox(lines=6, label="Prompt",
146
  value="Render a clear and concise summary of the photo.")
147
  with gr.Row():
148
  submit = gr.Button("Submit", variant="primary")
149
  clear = gr.Button("Clear", variant="secondary")
150
  with gr.Column():
151
+ output = gr.Textbox(label="Output", lines=8)
152
  data = gr.Dataset(components=[image, prompt], samples=examples, label="Examples", headers=["Image", "Prompt"])
153
  submit.click(chatbot, [image, prompt], [output])
154
  clear.click(lambda: None, [], [output])