Hugging Face's logo

Spaces:
hugginglearners
/
Ethiopian-Food-Classifier
Runtime error

App Files Community
Ethiopian-Food-Classifier
File size: 4,882 Bytes 
fd1e456
 
 
 
 
 
 
 
 
 
 
 
 
45b1522
fd1e456
 
 
424f496
fd1e456
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
 
import gradio as gr
from huggingface_hub import from_pretrained_fastai
from fastai.vision.all import *

repo_id = "Tinsae/EthioFoodtest3"

learn = from_pretrained_fastai(repo_id)
labels = learn.dls.vocab
EXAMPLES_PATH = Path('./examples')

title = "Ethiopian Food classifier "
description = """
This app is a demo of a model trained to classify images of the following Ethiopian food categories
- Beyaynetu, Chechebsa, Doro wat, Firfir, Genfo, Kikil, Kitfo, Shekla tibs, Shiro wat, Tihlo and Tire_siga

"""

article = "Full report on this model can be found [here](https://wandb.ai/tinsae/Ethiopian-foods/reports/Ethiopian-Foods-Classification---VmlldzoyMzExNjk1?accessToken=hx3g5jwmlrn059f11zp5v2ktg62ygl23mkxy2tevliu6bmqsmpazp5jkmqzjrg71)"
examples = [f'{EXAMPLES_PATH}/{f.name}' for f in EXAMPLES_PATH.iterdir()]

labels = learn.dls.vocab

v ='''<!DOCTYPE html>
      <html>
      <body>
      <h1>A recipe video</h1>
      {0}
      </body>
      </html> '''
v_ls = ['''<iframe width="956" height="240" src="https://www.youtube.com/embed/53Kj_dpBYzo" title="How to Make Ethiopian Food: Yetsom Beyaynetu | α‹¨α…αˆ α‰ α‹«α‹­αŠα‰± αŠ₯ሡራር" frameborder="0" allow="accelerometer; autoplay; clipboard-write; encrypted-media; gyroscope; picture-in-picture" allowfullscreen></iframe>''',
          '''<iframe width="956" height="240" src="https://www.youtube.com/embed/Hw9K9Q9Q_0M" title="Chechebsa (Spicy Flat Bread) | Ethiopian/Eritrean Food BY HabeshChef" frameborder="0" allow="accelerometer; autoplay; clipboard-write; encrypted-media; gyroscope; picture-in-picture" allowfullscreen></iframe>''',
          ''' <iframe width="956" height="240" src="https://www.youtube.com/embed/zi4AT6uYKUs" title="How To Make Spicy Ethiopian Chicken Stew: Doro Wat" frameborder="0" allow="accelerometer; autoplay; clipboard-write; encrypted-media; gyroscope; picture-in-picture" allowfullscreen></iframe>''',
          '''<iframe width="956" height="240" src="https://www.youtube.com/embed/V2fuMDmjqbc" title="Best Ethiopian Breakfast Food: Firfir Recipe (ፍርፍር አሰራር )" frameborder="0" allow="accelerometer; autoplay; clipboard-write; encrypted-media; gyroscope; picture-in-picture" allowfullscreen></iframe>''',
          '''<iframe width="956" height="240" src="https://www.youtube.com/embed/9iFMR3299BY" title="Genfo (Ethiopian style porridge)" frameborder="0" allow="accelerometer; autoplay; clipboard-write; encrypted-media; gyroscope; picture-in-picture" allowfullscreen></iframe>''', 
          '''<iframe width="956" height="240" src="https://www.youtube.com/embed/XtS1uzHGBYQ" title="Ethiopian Food EASY KIKIL Recipe α‹¨α‰…α‰…αˆ አሰራር | Amena and Elias" frameborder="0" allow="accelerometer; autoplay; clipboard-write; encrypted-media; gyroscope; picture-in-picture" allowfullscreen></iframe>''' ,
          '''<iframe width="956" height="240" src="https://www.youtube.com/embed/-xFR_daRIyw" title="Get a Taste Of Ethiopia: Kitfo with Woinee Mariam | Food Network" frameborder="0" allow="accelerometer; autoplay; clipboard-write; encrypted-media; gyroscope; picture-in-picture" allowfullscreen></iframe>''',
          '''<iframe width="956" height="240" src="https://www.youtube.com/embed/SFEu7JcQsis" title="Let’s cook Ethiopian Shekla Tibs! πŸ”₯" frameborder="0" allow="accelerometer; autoplay; clipboard-write; encrypted-media; gyroscope; picture-in-picture" allowfullscreen></iframe>''',
          '''<iframe width="956" height="240" src="https://www.youtube.com/embed/_W_J6wNq3nI" title="LET'S MAKE SHIRO TOGETHER AT HOME Vlogmas Day 22" frameborder="0" allow="accelerometer; autoplay; clipboard-write; encrypted-media; gyroscope; picture-in-picture" allowfullscreen></iframe>''',
          '''<iframe width="956" height="240" src="https://www.youtube.com/embed/7kDe2yBgB1I" title="α‰£αˆ…αˆ‹α‹Š የ αŒ₯αˆ•αˆŽ አሰራር | Traditional Tigrayan Tihlo Recipe | Ethiopian Food" frameborder="0" allow="accelerometer; autoplay; clipboard-write; encrypted-media; gyroscope; picture-in-picture" allowfullscreen></iframe>''',
          '''<iframe width="956" height="240" src="https://www.youtube.com/embed/Qex-uPtbUD0" title="FEEDING RAW MEAT TO GERMAN DIASPORA" frameborder="0" allow="accelerometer; autoplay; clipboard-write; encrypted-media; gyroscope; picture-in-picture" allowfullscreen></iframe>'''
        ]

def predict(img):
    img = PILImage.create(img)
    pred, pred_w_idx, probs = learn.predict(img)
    
    labels_probs = {labels[i]: float(probs[i]) for i, _ in enumerate(labels)}
    
    
    return labels_probs, v.format(v_ls[pred_w_idx])



demo = gr.Interface(predict,
                   "image",
                    [gr.outputs.Label(num_top_classes=3), "html"],
                    examples= examples,
                    title=title,
                    description=description,
                    article=article)

demo.launch()