app

app.py

@@ -431,10 +431,11 @@ class FoodRecognizer:
                 **load_kwargs
             )
 
-            # Učitaj model
+            # Učitaj model (force safetensors for security)
             self.model = AutoModelForImageClassification.from_pretrained(
                 model_name,
                 torch_dtype=torch.float16 if self.device == "cuda" else torch.float32,
+                use_safetensors=True,  # Force safetensors format (security)
                 **load_kwargs
             )
 
@@ -461,6 +462,7 @@ class FoodRecognizer:
                 )
                 self.model = AutoModelForImageClassification.from_pretrained(
                     model_name,
+                    use_safetensors=True,  # Force safetensors format (security)
                     **load_kwargs
                 )
                 self.model = self.model.to(self.device)

requirements.txt

@@ -4,9 +4,9 @@
 # ==================== Core Dependencies ====================
 # Python 3.9+ required
 
-# Deep Learning Framework
-torch>=2.0.0,<2.3.0
-torchvision>=0.15.0
+# Deep Learning Framework (upgraded for security CVE-2025-32434)
+torch>=2.6.0
+torchvision>=0.20.0
 
 # Hugging Face Transformers (for pretrained models)
 transformers>=4.35.0