Spaces:
Sleeping
Sleeping
NikiTricky
commited on
Commit
·
d1a8798
1
Parent(s):
985791e
Update app.py
Browse files
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)
|