Update app.py

app.py

@@ -9,7 +9,9 @@ from huggingface_hub import snapshot_download
 # MULTI-TURN MODEL SETUP
 # ---------------------------------
 MODEL_BASE_MULTI = snapshot_download(repo_id="nvidia/audio-flamingo-3-chat")
-model_multi = llava.load(MODEL_BASE_MULTI, model_base=None, devices=[0])
+# model_multi = llava.load(MODEL_BASE_MULTI, model_base=None, devices=[0])
+model_multi = llava.load(MODEL_BASE_MULTI, model_base=None)
+model_multi = model_multi.to("cuda")
 generation_config_multi = model_multi.default_generation_config