Update app.py
Browse files
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',
|