Uploading food classifier demo

README.md

@@ -1,13 +1,19 @@
 ---
-title: Food Not Food Classifier
-emoji: 🍔
-colorFrom: yellow
-colorTo: red
+title: Food Not Food Text Classifier
+emoji: 🍗🚫🥑
+colorFrom: blue
+colorTo: yellow
 sdk: gradio
-sdk_version: 4.44.1
+sdk_version: 4.36.1
 app_file: app.py
 pinned: false
+license: apache-2.0
 ---
 
-# Food Not Food Text Classifier
-A simple Gradio app to determine if a sentence is about food or not.
+# 🍗🚫🥑 Food Not Food Text Classifier
+
+Small demo to showcase a text classifier to determine if a sentence is about food or not food.
+
+DistillBERT model fine-tuned on a small synthetic dataset of 250 generated [Food or Not Food image captions](https://huggingface.co/datasets/mrdbourke/learn_hf_food_not_food_image_captions).
+
+[Source code notebook](https://github.com/mrdbourke/learn-huggingface/blob/main/notebooks/hugging_face_text_classification_tutorial.ipynb).