Update main.py

main.py

@@ -144,10 +144,10 @@ def interference(example, page_number=0):
         return image, table
     except IndexError as e:
         # Return a custom HTML-styled error message if an IndexError occurs
-        return f"<div style='color: grey; font-weight: bold;'>Error: in the current version of the model, the maximum number of words per page is 512.</div>"
+        gr.Error("Error: in the current version of the model, the maximum number of words per page is 512.")
     except Exception as e:
         # Handle other exceptions
-        return f"<div style='color: grey; font-weight: bold;'>An error occurred: {str(e)}</div>"
+       gr.Error(f"An error occurred: {e}")
 
 
 import gradio as gr