Update app.py

app.py

@@ -297,14 +297,14 @@ def extract(image):
         # im=im.convert("L")
         # im.save("demo.jpeg")
         # print("#### Image Saved #######")
-        #imageio.imwrite('demo.png',line_image)
+        imageio.imwrite('demo.jpeg',line_image)
 
 
 
-        #image = Image.open("demo.jpeg").convert("RGB")
+        image = Image.open("demo.jpeg").convert("RGB")
         
         #print("Started Processing")
-        image = line_image
+        #image = line_image
         pixel_values = processor(images=image, return_tensors="pt").pixel_values
 
         generated_ids = model.generate(pixel_values)
@@ -452,11 +452,11 @@ def extract_eval(image1,image2,image3,image4):
     print(a)
     text_sim_score=b
     model = ResNet50(include_top=False, weights='imagenet', pooling='avg')
-    # diagram_1_embed=return_image_embedding(model,image2)
-    # diagram_2_embed=return_image_embedding(model,image4)
-    # diagram_embed_sim_score=util.pytorch_cos_sim(embedding_1, embedding_2)
-    # print("Diagram Embedding Similarity Score \n")
-    # print(diagram_embed_sim_score)
+    diagram_1_embed=return_image_embedding(model,image2)
+    diagram_2_embed=return_image_embedding(model,image4)
+    diagram_embed_sim_score=util.pytorch_cos_sim(embedding_1, embedding_2)
+    print("Diagram Embedding Similarity Score \n")
+    print(diagram_embed_sim_score)