Update app.py

app.py

@@ -1,8 +1,15 @@
 import gradio as gr
+from huggingface_hub import from_pretrained_keras
 
-def greet(name):
-    return "Hello " + name + "!"
+model = from_pretrained_keras("NikiTricky/resnet50-food101")
 
-demo = gr.Interface(fn=greet, inputs="text", outputs="text")
+def classify_image(inp):
+  inp = inp.reshape((-1, 224, 224, 3))
+  inp /= 255.0
+  prediction = model.predict(inp)
+  confidences = {model.config['id2label'][str(i)]: float(prediction[i]) for i in range(101)}
+  return confidences
 
-demo.launch()   
+gr.Interface(fn=classify_image, 
+             inputs=gr.Image(shape=(224, 224)),
+             outputs=gr.Label(num_top_classes=3)).launch()