Update app.py

app.py

@@ -28,7 +28,8 @@ for model_key, config in MODEL_CONFIGS.items():
     print(f"Loading {model_key}...")
     models[model_key] = config["class"].from_pretrained(
         config["name"],
-        torch_dtype="auto"
+        torch_dtype="auto",
+        device_map="auto" 
     )
     processors[model_key] = AutoProcessor.from_pretrained(config["name"])