Update src/ai_processor.py

src/ai_processor.py

@@ -181,7 +181,7 @@ class AIProcessor:
                 self.models_cache['medgemma_pipe'] = pipeline(
                     'image-text-to-text',
                     model='google/medgemma-4b-it',
-                    device='cuda',
+                    device='auto',
                     torch_dtype=torch.bfloat16,
                     offload_folder='offload',
                     token=self.config.HF_TOKEN