burman-ai commited on
Commit
78c283c
Β·
verified Β·
1 Parent(s): 0cc86f5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -10,7 +10,7 @@ def food_not_food_classifier(text: str) -> Dict[str, float]:
10
  # Set up text classification pipeline
11
  food_not_food_classifier = pipeline(task="text-classification",
12
  # Because our model is on Hugging Face already, we can pass in the model name directly
13
- model="burman-ai/DistilBert-base-uncased", # link to model on HF Hub
14
  device="cuda" if torch.cuda.is_available() else "cpu",
15
  top_k=None) # return all possible scores (not just top-1)
16
 
 
10
  # Set up text classification pipeline
11
  food_not_food_classifier = pipeline(task="text-classification",
12
  # Because our model is on Hugging Face already, we can pass in the model name directly
13
+ model="burman-ai/bert-base-uncased", # link to model on HF Hub
14
  device="cuda" if torch.cuda.is_available() else "cpu",
15
  top_k=None) # return all possible scores (not just top-1)
16