Update app.py

app.py

@@ -1,6 +1,6 @@
 import re
 import gradio as gr
-print("gradio version:", gr.__version__)
+import gradio_theme
 
 import torch
 from transformers import DonutProcessor, VisionEncoderDecoderModel
@@ -49,21 +49,6 @@ def process_document(image):
 #description = "Clip AI: Check Understanding"
 #article = "<p style='text-align: center'><a href='https://arxiv.org/abs/2111.15664' target='_blank'>Donut: OCR-free Document Understanding Transformer</a> | <a href='https://github.com/clovaai/donut' target='_blank'>Github Repo</a></p>"
 
-# Style
-primary_palette = gr.themes.Color(
-        c50="#282253",
-        c100="#282253",
-        c200="#282253",
-        c300="#282253",
-        c400="#282253",
-        c500="#282253",
-        c600="white",
-        c700="#EC5F9C",
-        c800="#EC4D92",
-        c900="#EB3A86",
-        c950='#E9277A',
-    )
-
 demo = gr.Interface(
     fn=process_document,
     inputs="image",
@@ -74,9 +59,7 @@ demo = gr.Interface(
     enable_queue=True,
     #examples=[["example.png"], ["example_2.png"], ["example_3.png"]],
     cache_examples=False,
-    theme=gr.themes.Default(
-        primary_hue=primary_palette
-    ).set(body_background_fill='#F8F9FA')
+    theme=gradio_theme.theme
 )
 
 demo.launch()