IncreasingLoss commited on
Commit
4cda534
·
verified ·
1 Parent(s): af28535

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +17 -0
app.py CHANGED
@@ -111,6 +111,23 @@ model = MobileNetV3_Small().cpu()
111
  model.load_state_dict(torch.load("MobileNet3_small_StateDictionary.pth", map_location='cpu'))
112
  model.eval()
113
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
114
  # Class Labels
115
  classes = [
116
  'antelope', 'buffalo', 'chimpanzee', 'cow', 'deer', 'dolphin',
 
111
  model.load_state_dict(torch.load("MobileNet3_small_StateDictionary.pth", map_location='cpu'))
112
  model.eval()
113
 
114
+
115
+ css = """
116
+ .centered-examples {
117
+ margin: 0 auto !important;
118
+ justify-content: center !important;
119
+ gap: 8px !important;
120
+ min-height: 150px !important; /* Added minimum height */
121
+ }
122
+ .centered-examples .thumb {
123
+ height: 100px !important;
124
+ width: 100px !important;
125
+ object-fit: cover !important;
126
+ margin: 5px !important; /* Added margin between thumbs */
127
+ }
128
+ """
129
+
130
+
131
  # Class Labels
132
  classes = [
133
  'antelope', 'buffalo', 'chimpanzee', 'cow', 'deer', 'dolphin',