Update app.py
Browse files
app.py
CHANGED
|
@@ -1,6 +1,6 @@
|
|
| 1 |
#!/usr/bin/env python3
|
| 2 |
"""
|
| 3 |
-
|
| 4 |
"""
|
| 5 |
|
| 6 |
import gradio as gr
|
|
@@ -177,7 +177,7 @@ except Exception as e:
|
|
| 177 |
# ============================================================================
|
| 178 |
|
| 179 |
def predict(image):
|
| 180 |
-
"""Predict ImageNet class for input image
|
| 181 |
|
| 182 |
if image is None:
|
| 183 |
return {
|
|
@@ -256,7 +256,7 @@ print("Model ready!")
|
|
| 256 |
|
| 257 |
with gr.Blocks(theme=gr.themes.Soft()) as demo:
|
| 258 |
gr.Markdown("""
|
| 259 |
-
# 🔥 ImageNet ResNet50 Classifier
|
| 260 |
|
| 261 |
**77.09% Top-1 Accuracy** - From scratch training
|
| 262 |
|
|
@@ -276,7 +276,7 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
|
|
| 276 |
gr.Markdown("""
|
| 277 |
**Model:** ResNet50 (25.5M params) | **Accuracy:** 77.09%
|
| 278 |
|
| 279 |
-
[GitHub](https://github.com/Shwethaamrutha/TSAI-
|
| 280 |
""")
|
| 281 |
|
| 282 |
if __name__ == "__main__":
|
|
|
|
| 1 |
#!/usr/bin/env python3
|
| 2 |
"""
|
| 3 |
+
HuggingFace App for ImageNet ResNet50
|
| 4 |
"""
|
| 5 |
|
| 6 |
import gradio as gr
|
|
|
|
| 177 |
# ============================================================================
|
| 178 |
|
| 179 |
def predict(image):
|
| 180 |
+
"""Predict ImageNet class for input image """
|
| 181 |
|
| 182 |
if image is None:
|
| 183 |
return {
|
|
|
|
| 256 |
|
| 257 |
with gr.Blocks(theme=gr.themes.Soft()) as demo:
|
| 258 |
gr.Markdown("""
|
| 259 |
+
# 🔥 ImageNet ResNet50 Classifier
|
| 260 |
|
| 261 |
**77.09% Top-1 Accuracy** - From scratch training
|
| 262 |
|
|
|
|
| 276 |
gr.Markdown("""
|
| 277 |
**Model:** ResNet50 (25.5M params) | **Accuracy:** 77.09%
|
| 278 |
|
| 279 |
+
[GitHub](https://github.com/Shwethaamrutha/TSAI-S9)
|
| 280 |
""")
|
| 281 |
|
| 282 |
if __name__ == "__main__":
|