Update app.py

app.py

@@ -267,7 +267,8 @@ parser.add_argument(
     help="The maximum number of new generated tokens.",
 )
 args = parser.parse_args()
-
+a=torch.tensor(1).cuda()
+print(a)
 model = EaModel.from_pretrained(
     base_model_path=args.base_model_path,
     ea_model_path=args.ea_model_path,