jianjiang
/
vit-age-classifier
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

replace theme
Build-Deploy-Actions Details

This commit is contained in:
jianjiang 2023-04-21 17:09:50 +08:00
parent b93b9dba15
commit b6ffc33c64
1 changed files with 10 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
app.py
View File

@ -2,6 +2,15 @@ from PIL import Image
import gradio as gr import gradio as gr
from transformers import ViTFeatureExtractor, ViTForImageClassification from transformers import ViTFeatureExtractor, ViTForImageClassification
import torch 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 # Init model, transforms
model = ViTForImageClassification.from_pretrained('nateraw/vit-age-classifier') model = ViTForImageClassification.from_pretrained('nateraw/vit-age-classifier')
@ -43,5 +52,5 @@ gr.Interface(
description=description, description=description,
article=article, article=article,
examples=examples, examples=examples,
theme="huggingface", theme=theme,
).launch(debug=True, enable_queue=True) ).launch(debug=True, enable_queue=True)