dhairyashil commited on
Commit
64a2bca
·
1 Parent(s): 0349841

update acc info

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -66,14 +66,14 @@ def predict(image):
66
 
67
  # Create Gradio interface
68
  title = "ImageNet1k Classification"
69
- description = """Upload an image and the model will predict its category using the ImageNet1k classification system.
70
  Tips for best results:
71
  - Use clear, well-lit images; ensure the main subject is centered and clearly visible
72
  - The model works best with common objects, animals, and scenes
73
  - Images can be any size or color - they'll be automatically resized to 224x224
74
 
75
  The model will show the top 5 most likely categories with confidence scores.
76
- Link to github repo: [https://github.com/dhairyag/ImageNet1k_ResNet50](https://github.com/dhairyag/ImageNet1k_ResNet50)
77
  """
78
 
79
  iface = gr.Interface(
 
66
 
67
  # Create Gradio interface
68
  title = "ImageNet1k Classification"
69
+ description = """Accuracy for this model is top@1: 75.212%, top@5: 92.668% after 80 epochs. Upload an image and the model will predict its category using the ImageNet1k classification system.
70
  Tips for best results:
71
  - Use clear, well-lit images; ensure the main subject is centered and clearly visible
72
  - The model works best with common objects, animals, and scenes
73
  - Images can be any size or color - they'll be automatically resized to 224x224
74
 
75
  The model will show the top 5 most likely categories with confidence scores.
76
+ Link to github repo: [https://github.com/dhairyag/ImageNet1k_ResNet50](https://github.com/dhairyag/ImageNet1k_ResNet50)
77
  """
78
 
79
  iface = gr.Interface(