joadithya commited on
Commit
61f8b0e
·
verified ·
1 Parent(s): 305ba48

fix: app.py gradio demo

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -33,6 +33,7 @@ Fine-tuned from [DistilBERT](https://huggingface.co/distilbert/distilbert-base-u
33
  """
34
 
35
  demo = gr.Interface(
 
36
  inputs="text",
37
  outputs=gr.Label(num_top_classes=2),
38
  title="Food Caption Classifier",
 
33
  """
34
 
35
  demo = gr.Interface(
36
+ fn=food_not_food_classifier,
37
  inputs="text",
38
  outputs=gr.Label(num_top_classes=2),
39
  title="Food Caption Classifier",