mscsasem3 commited on
Commit
a6cb655
1 Parent(s): 78abc18

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -27,6 +27,7 @@ import os
27
  from tensorflow.keras.applications.resnet50 import ResNet50,preprocess_input, decode_predictions
28
  from tensorflow.keras.preprocessing import image
29
  from sklearn.feature_extraction.text import TfidfVectorizer
 
30
 
31
 
32
 
@@ -465,7 +466,7 @@ def extract_eval(image1,image2,image3,image4):
465
  diagram_2_embed=return_image_embedding(model,image4)
466
  diagram_embed_sim_score=util.pytorch_cos_sim(diagram_1_embed, diagram_2_embed)
467
  print("Diagram Embedding Similarity Score \n")
468
- print(diagram_embed_sim_score)
469
  diagram_1_text=inference(image2,'en')
470
  diagram_2_text=inference(image4,'en')
471
  print(diagram_1_text)
 
27
  from tensorflow.keras.applications.resnet50 import ResNet50,preprocess_input, decode_predictions
28
  from tensorflow.keras.preprocessing import image
29
  from sklearn.feature_extraction.text import TfidfVectorizer
30
+ import easyocr
31
 
32
 
33
 
 
466
  diagram_2_embed=return_image_embedding(model,image4)
467
  diagram_embed_sim_score=util.pytorch_cos_sim(diagram_1_embed, diagram_2_embed)
468
  print("Diagram Embedding Similarity Score \n")
469
+ print(str(int(float(str(diagram_embed_sim_score).split("[")[2].split("]")[0])*10.0)))
470
  diagram_1_text=inference(image2,'en')
471
  diagram_2_text=inference(image4,'en')
472
  print(diagram_1_text)