prithivMLmods commited on
Commit
af2318a
·
verified ·
1 Parent(s): fbf3598

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +58 -0
app.py CHANGED
@@ -1,6 +1,10 @@
1
  import gradio as gr
2
  import torch
 
3
  import spaces
 
 
 
4
  from transformers import AutoModel, AutoProcessor
5
 
6
  from gender_classification import gender_classification
@@ -19,6 +23,60 @@ from rice_leaf_disease import classify_leaf_disease
19
  from traffic_density import traffic_density_classification
20
  from clip_art import clipart_classification
21
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
22
  # Main classification function for multi-model classification.
23
  def classify(image, model_name):
24
  if model_name == "gender":
 
1
  import gradio as gr
2
  import torch
3
+ import time
4
  import spaces
5
+ from typing import Iterable
6
+ from gradio.themes.base import Base
7
+ from gradio.themes.utils import colors, fonts, sizes
8
  from transformers import AutoModel, AutoProcessor
9
 
10
  from gender_classification import gender_classification
 
23
  from traffic_density import traffic_density_classification
24
  from clip_art import clipart_classification
25
 
26
+ #Gradio-Theme
27
+ class Seafoam(Base):
28
+ def __init__(
29
+ self,
30
+ *,
31
+ primary_hue: colors.Color | str = colors.emerald,
32
+ secondary_hue: colors.Color | str = colors.blue,
33
+ neutral_hue: colors.Color | str = colors.blue,
34
+ spacing_size: sizes.Size | str = sizes.spacing_md,
35
+ radius_size: sizes.Size | str = sizes.radius_md,
36
+ text_size: sizes.Size | str = sizes.text_lg,
37
+ font: fonts.Font
38
+ | str
39
+ | Iterable[fonts.Font | str] = (
40
+ fonts.GoogleFont("Quicksand"),
41
+ "ui-sans-serif",
42
+ "sans-serif",
43
+ ),
44
+ font_mono: fonts.Font
45
+ | str
46
+ | Iterable[fonts.Font | str] = (
47
+ fonts.GoogleFont("IBM Plex Mono"),
48
+ "ui-monospace",
49
+ "monospace",
50
+ ),
51
+ ):
52
+ super().__init__(
53
+ primary_hue=primary_hue,
54
+ secondary_hue=secondary_hue,
55
+ neutral_hue=neutral_hue,
56
+ spacing_size=spacing_size,
57
+ radius_size=radius_size,
58
+ text_size=text_size,
59
+ font=font,
60
+ font_mono=font_mono,
61
+ )
62
+ super().set(
63
+ body_background_fill="repeating-linear-gradient(45deg, *primary_200, *primary_200 10px, *primary_50 10px, *primary_50 20px)",
64
+ body_background_fill_dark="repeating-linear-gradient(45deg, *primary_800, *primary_800 10px, *primary_900 10px, *primary_900 20px)",
65
+ button_primary_background_fill="linear-gradient(90deg, *primary_300, *secondary_400)",
66
+ button_primary_background_fill_hover="linear-gradient(90deg, *primary_200, *secondary_300)",
67
+ button_primary_text_color="white",
68
+ button_primary_background_fill_dark="linear-gradient(90deg, *primary_600, *secondary_800)",
69
+ slider_color="*secondary_300",
70
+ slider_color_dark="*secondary_600",
71
+ block_title_text_weight="600",
72
+ block_border_width="3px",
73
+ block_shadow="*shadow_drop_lg",
74
+ button_primary_shadow="*shadow_drop_lg",
75
+ button_large_padding="32px",
76
+ )
77
+
78
+ seafoam = Seafoam()
79
+
80
  # Main classification function for multi-model classification.
81
  def classify(image, model_name):
82
  if model_name == "gender":