chats-bug commited on
Commit
bc65b96
1 Parent(s): 8ca734f

Removed typing

Browse files
Files changed (1) hide show
  1. app.py +11 -14
app.py CHANGED
@@ -5,9 +5,9 @@ from model import GitBaseCocoModel
5
 
6
 
7
  def generate_captions(
8
- image: Image,
9
- max_len: int = 50,
10
- num_captions: int = 1,
11
  ):
12
  """
13
  Generates captions for the given image.
@@ -36,22 +36,19 @@ def generate_captions(
36
  caption = "\n".join(caption)
37
  return caption
38
 
39
-
40
- inputs = [
41
- gr.inputs.Image(type="pil", label="Image"),
42
- gr.inputs.Slider(minimum=20, maximum=100, step=5, default=50, label="Maximum Caption Length"),
43
- gr.inputs.Slider(minimum=1, maximum=10, step=1, default=1, label="Number of Captions to Generate"),
44
- ]
45
- # Determine the number of outputs based on the number of captions to generate.
46
- outputs = gr.outputs.Textbox(label="Captions")
47
-
48
  title = "Git-Base-COCO Image Captioning"
49
  description = "A model for generating captions for images."
50
 
51
  interface = gr.Interface(
52
  fn=generate_captions,
53
- inputs=inputs,
54
- outputs=outputs,
 
 
 
 
 
 
55
  title=title,
56
  description=description,
57
  )
 
5
 
6
 
7
  def generate_captions(
8
+ image,
9
+ max_len,
10
+ num_captions,
11
  ):
12
  """
13
  Generates captions for the given image.
 
36
  caption = "\n".join(caption)
37
  return caption
38
 
 
 
 
 
 
 
 
 
 
39
  title = "Git-Base-COCO Image Captioning"
40
  description = "A model for generating captions for images."
41
 
42
  interface = gr.Interface(
43
  fn=generate_captions,
44
+ inputs=[
45
+ gr.inputs.Image(type="pil", label="Image"),
46
+ gr.inputs.Slider(minimum=20, maximum=100, step=5, default=50, label="Maximum Caption Length"),
47
+ gr.inputs.Slider(minimum=1, maximum=10, step=1, default=1, label="Number of Captions to Generate"),
48
+ ],
49
+ outputs=[
50
+ gr.outputs.Textbox(label="Caption"),
51
+ ],
52
  title=title,
53
  description=description,
54
  )