John6666 commited on
Commit
719ad71
·
verified ·
1 Parent(s): 238df0e

Upload 2 files

Browse files
Files changed (2) hide show
  1. app.py +3 -4
  2. 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", format="png", label="Upload an Image")
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