Uploading food not food text classifier demo app

app.py

@@ -7,6 +7,22 @@ from transformers import pipeline
 
 huggingface_model_path = "Suraj-Yadav/learn_hf_food_not_food_text_classifier-distilbert-base-uncased"
 
+
+def set_device() -> torch.device:
+    """
+    Set the device to the best available option: CUDA (if available), MPS (if available on Mac),
+    or CPU as a fallback. Provides a robust selection mechanism for production environments.
+
+    Returns:
+        torch.device: The best available device for computation.
+    """
+    if torch.cuda.is_available():
+        return torch.device("cuda")
+    elif torch.backends.mps.is_available() and torch.backends.mps.is_built():
+        return torch.device("mps")
+    else:
+        return torch.device("cpu")
+
 # 2. Define function to use our model on given text 
 def food_not_food_classifier(
     text: Union[str, list],