gschurck commited on
Commit
3c4d86c
·
1 Parent(s): 325991d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -9,11 +9,11 @@ def get_image_embedding(image):
9
 
10
  # Load and preprocess the image
11
  img_emb = image_model.encode(image)
12
-
13
  return {"prediction": img_emb}
14
 
15
 
16
  image_input = gr.Image(type="pil")
17
- label_output = gr.outputs.Label()
18
 
19
  gr.Interface(fn=get_image_embedding, inputs=image_input, outputs=label_output).launch()
 
9
 
10
  # Load and preprocess the image
11
  img_emb = image_model.encode(image)
12
+ print(img_emb)
13
  return {"prediction": img_emb}
14
 
15
 
16
  image_input = gr.Image(type="pil")
17
+ label_output = gr.outputs.JSON()
18
 
19
  gr.Interface(fn=get_image_embedding, inputs=image_input, outputs=label_output).launch()