fffiloni commited on
Commit
00856ac
1 Parent(s): b240a22

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -7,7 +7,7 @@ from gradio_client import Client
7
  #fuyu_client = Client("https://adept-fuyu-8b-demo.hf.space/")
8
  kosmos2_client = Client("https://ydshieh-kosmos-2.hf.space/")
9
 
10
- def get_caption_from_kosmos2(image_in):
11
  """
12
  fuyu_result = fuyu_client.predict(
13
  image_in, # str representing input in 'raw_image' Image component
@@ -98,7 +98,7 @@ instruction = f"""
98
  """
99
 
100
  def infer(image_in):
101
- gr.Info("Getting image caption with moondream1...")
102
  user_prompt = get_caption(image_in)
103
 
104
  prompt = f"{instruction.strip()}\n{user_prompt}</s>"
@@ -143,7 +143,7 @@ with gr.Blocks(css=css) as demo:
143
  submit_btn = gr.Button("Make LLM system from my pic !")
144
  with gr.Column():
145
  caption = gr.Textbox(
146
- label = "Image caption (moondream1)",
147
  elem_id = "image-caption"
148
  )
149
  result = gr.Textbox(
 
7
  #fuyu_client = Client("https://adept-fuyu-8b-demo.hf.space/")
8
  kosmos2_client = Client("https://ydshieh-kosmos-2.hf.space/")
9
 
10
+ def get_caption(image_in):
11
  """
12
  fuyu_result = fuyu_client.predict(
13
  image_in, # str representing input in 'raw_image' Image component
 
98
  """
99
 
100
  def infer(image_in):
101
+ gr.Info("Getting image caption with Kosmos2...")
102
  user_prompt = get_caption(image_in)
103
 
104
  prompt = f"{instruction.strip()}\n{user_prompt}</s>"
 
143
  submit_btn = gr.Button("Make LLM system from my pic !")
144
  with gr.Column():
145
  caption = gr.Textbox(
146
+ label = "Image caption (Kosmos2)",
147
  elem_id = "image-caption"
148
  )
149
  result = gr.Textbox(