Update app.py

app.py

@@ -69,7 +69,10 @@ for key,value in finetuned_path.items():
     if 'transformer.base_model.model.' in key:
         transformer_dict[key.replace('transformer.base_model.model.', '')] = value 
 pipeline.transformer.load_state_dict(transformer_dict, strict=False)
-pipeline.to('cuda')
+#pipeline.to('cuda')
+# Tự động kiểm tra: Có GPU thì dùng, không có thì dùng CPU
+device = "cuda" if torch.cuda.is_available() else "cpu"
+pipeline.to(device)
 pipeline.enable_vae_slicing()
 pipeline.enable_vae_tiling()