Update app.py

app.py

@@ -52,16 +52,21 @@ if __name__ == "__main__":
         sentence_f1 = calculate_F1_for_spans(tree_to_spans(gold_standard), tree_to_spans(best_parse))
         TreeView(Tree.fromstring(gold_standard))._cframe.print_to_file('gold_standard.ps')
         TreeView(Tree.fromstring(best_parse))._cframe.print_to_file('best_parse.ps')
-        os.system('convert gold_standard.ps gold_standard.png')
-        os.system('convert best_parse.ps best_parse.png')
-        print(os.listdir())
-        gold_standard_img = Image.open("gold_standard.png")
-        best_parse_img = Image.open("best_parse.png")
-        #gold_standard_img.save('gold_standard.png')
-        #best_parse_img.save('best_parse.png')
-        #gold_standard_img_png = Image.open("gold_standard.png")
-        #best_parse_img_png = Image.open("best_parse.png")
-        return gold_standard_img, best_parse_img, f"{sentence_f1:.2f}"
+        #os.system('convert gold_standard.ps gold_standard.png')
+        #os.system('convert best_parse.ps best_parse.png')
+        #print(os.listdir())
+        gold_standard_img = Image.open("gold_standard.ps")
+        best_parse_img = Image.open("best_parse.ps")
+        print(gold_standard_img.info)
+        print(best_parse_img.info)
+        
+        gold_standard_img.save('gold_standard.png', dpi=gold_standard_img.info['dpi'])
+        best_parse_img.save('best_parse.png', dpi=best_parse_img.info['dpi'])
+        
+        gold_standard_img_png = Image.open("gold_standard.png")
+        best_parse_img_png = Image.open("best_parse.png")
+        
+        return gold_standard_img_png, best_parse_img_png, f"{sentence_f1:.2f}"
 
 
     iface = gradio.Interface(