From b6ffc33c64ff147d56fff9948a3fc3fcf5c52da1 Mon Sep 17 00:00:00 2001
From: jianjiang <jianjiang@iflytek.com>
Date: Fri, 21 Apr 2023 17:09:50 +0800
Subject: [PATCH] replace theme

---
 app.py | 11 ++++++++++-
 1 file changed, 10 insertions(+), 1 deletion(-)

diff --git a/app.py b/app.py
index 1688b9d..918931f 100644
--- a/app.py
+++ b/app.py
@@ -2,6 +2,15 @@ from PIL import Image
 import gradio as gr
 from transformers import ViTFeatureExtractor, ViTForImageClassification
 import torch
+from gradio.themes.utils import sizes
+theme = gr.themes.Default(radius_size=sizes.radius_none).set(
+    block_label_text_color = '#4D63FF',
+    block_title_text_color = '#4D63FF',
+    button_primary_text_color = '#4D63FF',
+    button_primary_background_fill='#FFFFFF',
+    button_primary_border_color='#4D63FF',
+    button_primary_background_fill_hover='#EDEFFF',
+)
 
 # Init model, transforms
 model = ViTForImageClassification.from_pretrained('nateraw/vit-age-classifier')
@@ -43,5 +52,5 @@ gr.Interface(
     description=description,
     article=article,
     examples=examples,
-    theme="huggingface",
+    theme=theme,
 ).launch(debug=True, enable_queue=True)