Uploading our food not food classifier demo from a notebook!
Browse files
app.py
CHANGED
@@ -1,11 +1,11 @@
|
|
1 |
-
# 1.
|
2 |
import torch
|
3 |
import gradio as gr
|
4 |
|
5 |
from typing import Dict
|
6 |
from transformers import pipeline
|
7 |
|
8 |
-
# 2.
|
9 |
food_not_food_classifier = pipeline(task="text-classification",
|
10 |
model="tonicanada/learn_hf_food_not_food_text_classifier-distilbert-base-uncased",
|
11 |
top_k=1,
|
|
|
1 |
+
# 1. Importamos las librer铆as necesarias
|
2 |
import torch
|
3 |
import gradio as gr
|
4 |
|
5 |
from typing import Dict
|
6 |
from transformers import pipeline
|
7 |
|
8 |
+
# 2. Deifnimos nuestra funci贸n para usar el modelo
|
9 |
food_not_food_classifier = pipeline(task="text-classification",
|
10 |
model="tonicanada/learn_hf_food_not_food_text_classifier-distilbert-base-uncased",
|
11 |
top_k=1,
|