shwethd commited on
Commit
551fdaf
·
verified ·
1 Parent(s): 106a88d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -1,6 +1,6 @@
1
  #!/usr/bin/env python3
2
  """
3
- DEBUG VERSION - HuggingFace App for ImageNet ResNet50
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 - WITH DEBUG INFO"""
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 (DEBUG VERSION)
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-S8)
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__":