Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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/
|
| 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 |
|