Update app.py

app.py

@@ -178,14 +178,14 @@ def layoutjson2md(image: Image.Image, layout_data: List[Dict], text_key: str = '
                             buffer = BytesIO()
                             cropped_img.save(buffer, format='PNG')
                             img_data = base64.b64encode(buffer.getvalue()).decode()
-                            markdown_lines.append(f"<image-card alt="Image" src="data:image/png;base64,{img_data}" ></image-card>\n")
+                            markdown_lines.append(f"![Image](data:image/png;base64,{img_data})\n")
                         else:
-                            markdown_lines.append("<image-card alt="Image" src="Image region detected" ></image-card>\n")
+                            markdown_lines.append("![Image](Image region detected)\n")
                     except Exception as e:
                         print(f"Error processing image region: {e}")
-                        markdown_lines.append("<image-card alt="Image" src="Image detected" ></image-card>\n")
+                        markdown_lines.append("![Image](Image detected)\n")
                 else:
-                    markdown_lines.append("<image-card alt="Image" src="Image detected" ></image-card>\n")
+                    markdown_lines.append("![Image](Image detected)\n")
             elif not text:
                 continue
             elif category == 'Title':
@@ -203,7 +203,7 @@ def layoutjson2md(image: Image.Image, layout_data: List[Dict], text_key: str = '
                     markdown_lines.append(f"**Table:** {text}\n")
             elif category == 'Formula':
                 if text.strip().startswith('$') or '\\' in text:
-                    markdown_lines.append(f"$$  \n{text}\n  $$\n")
+                    markdown_lines.append(f"$$\n{text}\n$$\n")
                 else:
                     markdown_lines.append(f"**Formula:** {text}\n")
             elif category == 'Caption':
@@ -220,6 +220,7 @@ def layoutjson2md(image: Image.Image, layout_data: List[Dict], text_key: str = '
         return str(layout_data)
     return "\n".join(markdown_lines)
 
+
 @spaces.GPU
 def inference(model_name: str, image: Image.Image, text: str, max_new_tokens: int = 1024) -> str:
     try: