Spaces:
Sleeping
Sleeping
howard-hou
commited on
Commit
•
922f821
1
Parent(s):
fbe0b0a
Update app.py
Browse files
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=
|
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=
|
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])
|