Spaces:
Runtime error
Runtime error
Upload 2 files
Browse files- app.py +3 -4
- requirements.txt +2 -1
app.py
CHANGED
@@ -34,7 +34,6 @@ moondream = AutoModelForCausalLM.from_pretrained(
|
|
34 |
).to(device)
|
35 |
moondream.eval()
|
36 |
|
37 |
-
|
38 |
@spaces.GPU
|
39 |
def answer_question(img, prompt):
|
40 |
image_embeds = moondream.encode_image(img)
|
@@ -96,13 +95,13 @@ with gr.Blocks() as demo:
|
|
96 |
prompt = gr.Textbox(label="Input", value="Describe this image.", scale=4)
|
97 |
submit = gr.Button("Submit")
|
98 |
with gr.Row():
|
99 |
-
img = gr.Image(type="pil", image_mode="RGB",
|
100 |
with gr.Column():
|
101 |
output = gr.Markdown(label="Response")
|
102 |
ann = gr.Image(visible=False, label="Annotated Image")
|
103 |
|
104 |
-
submit.click(answer_question, [img, prompt], output)
|
105 |
-
prompt.submit(answer_question, [img, prompt], output)
|
106 |
output.change(process_answer, [img, output], ann, show_progress=False)
|
107 |
|
108 |
demo.queue().launch()
|
|
|
34 |
).to(device)
|
35 |
moondream.eval()
|
36 |
|
|
|
37 |
@spaces.GPU
|
38 |
def answer_question(img, prompt):
|
39 |
image_embeds = moondream.encode_image(img)
|
|
|
95 |
prompt = gr.Textbox(label="Input", value="Describe this image.", scale=4)
|
96 |
submit = gr.Button("Submit")
|
97 |
with gr.Row():
|
98 |
+
img = gr.Image(type="pil", image_mode="RGB", label="Upload an Image")
|
99 |
with gr.Column():
|
100 |
output = gr.Markdown(label="Response")
|
101 |
ann = gr.Image(visible=False, label="Annotated Image")
|
102 |
|
103 |
+
submit.click(answer_question, [img, prompt], output, queue=True)
|
104 |
+
prompt.submit(answer_question, [img, prompt], output, queue=True)
|
105 |
output.change(process_answer, [img, output], ann, show_progress=False)
|
106 |
|
107 |
demo.queue().launch()
|
requirements.txt
CHANGED
@@ -3,4 +3,5 @@ transformers>=4.44.0
|
|
3 |
einops>=0.8.0
|
4 |
accelerate>=0.32.1
|
5 |
numpy<2
|
6 |
-
torch==2.4.0
|
|
|
|
3 |
einops>=0.8.0
|
4 |
accelerate>=0.32.1
|
5 |
numpy<2
|
6 |
+
torch==2.4.0
|
7 |
+
torchvision
|