barabum commited on
Commit
2d0598e
1 Parent(s): b45d1aa
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -10,7 +10,7 @@ def image_classifier(im1: numpy.ndarray, im2: numpy.ndarray):
10
  encoded_image = model.encode([Image.fromarray(im1), Image.fromarray(im2)], batch_size=128,
11
  convert_to_tensor=True, show_progress_bar=True)
12
  processed_images = util.paraphrase_mining_embeddings(encoded_image)
13
- return {"Схожи на": round(processed_images[0][0]*100, 2)}
14
 
15
 
16
  with gr.Blocks() as b:
 
10
  encoded_image = model.encode([Image.fromarray(im1), Image.fromarray(im2)], batch_size=128,
11
  convert_to_tensor=True, show_progress_bar=True)
12
  processed_images = util.paraphrase_mining_embeddings(encoded_image)
13
+ return {"Схожи на": round(processed_images[0][0], 2)}
14
 
15
 
16
  with gr.Blocks() as b: