Spaces:

manasuma
/

FoodClassificationViT

Build error

manasuma commited on Jan 13, 2023

Commit

abe2b17

1 Parent(s): 2e29846

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -60,7 +60,7 @@ description = "A food classification model base on ViT"
 # Create Gradio interface
 demo = gr.Interface(
     fn=predict,
-    inputs=gr.Image,
     outputs=[
         gr.Label(num_top_classes=5, label="Predictions"),
         gr.Number(label="Prediction time (s)"),

 # Create Gradio interface
 demo = gr.Interface(
     fn=predict,
+    inputs=gr.Image(type="pil"),
     outputs=[
         gr.Label(num_top_classes=5, label="Predictions"),
         gr.Number(label="Prediction time (s)"),