Update app.py
Browse files
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)
|