simecek commited on
Commit
d05120e
·
verified ·
1 Parent(s): 5f594b2

Image size fixed as 224 x 224

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -14,7 +14,7 @@ def classify_image(img):
14
  return dict(zip(categories, map(float, probs))) # Map categories to their probabilities
15
 
16
  # Define Gradio input and output components
17
- image = gr.Image() # Image input
18
  label = gr.Label() # Output label to display classification
19
  examples = ['test_image1.jpg', 'test_image2.jpg'] # Example images for demonstration
20
 
 
14
  return dict(zip(categories, map(float, probs))) # Map categories to their probabilities
15
 
16
  # Define Gradio input and output components
17
+ image = gr.Image(width=224, height=224) # Image input
18
  label = gr.Label() # Output label to display classification
19
  examples = ['test_image1.jpg', 'test_image2.jpg'] # Example images for demonstration
20