Update app.py

app.py

@@ -20,7 +20,7 @@ base_model = AutoModelForCausalLM.from_pretrained(
 
 ckpts = "ckpts/Qlora_adaptor/"
 phi2_model = PeftModel.from_pretrained(base_model, ckpts)
-phi2_model = phi2_model.merge_and_unload().to(device)
+phi2_model = phi2_model.merge_and_unload().to(config.get("device"))
 
 projection_layer = torch.nn.Linear(config.get("clip_embed"), config.get("phi_embed"))
 projection_layer.load_state_dict(torch.load('./ckpts/model_phase2.pth', map_location=config.get("device")))