NikiTricky commited on
Commit
d1a8798
·
1 Parent(s): 985791e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -4,6 +4,7 @@ from huggingface_hub import from_pretrained_keras
4
  model = from_pretrained_keras("NikiTricky/resnet50-food101")
5
 
6
  def classify_image(inp):
 
7
  inp = inp.reshape((-1, 224, 224, 3))
8
  inp /= 255.0
9
  prediction = model.predict(inp)
 
4
  model = from_pretrained_keras("NikiTricky/resnet50-food101")
5
 
6
  def classify_image(inp):
7
+ print("Classifying image...")
8
  inp = inp.reshape((-1, 224, 224, 3))
9
  inp /= 255.0
10
  prediction = model.predict(inp)