Update app.py

app.py

@@ -6,14 +6,7 @@ from transformers import AutoModel, AutoProcessor
 from transformers import StoppingCriteria, TextIteratorStreamer, StoppingCriteriaList
 
 
-parser = argparse.ArgumentParser()
-# parser.add_argument("--device", type=str, default="cuda:0")
-# parser.add_argument("--ckpt_path", type=str, default="./salmonn_7b_v0.pth")
-# parser.add_argument("--whisper_path", type=str, default="./whisper_large_v2")
-# parser.add_argument("--beats_path", type=str, default="./beats/BEATs_iter3_plus_AS2M_finetuned_on_AS2M_cpt2.pt")
-# parser.add_argument("--vicuna_path", type=str, default="./vicuna-7b-v1.5")
-# parser.add_argument("--low_resource", action='store_true', default=False)
-parser.add_argument("--port", default=9527)
+device = "cuda:0" if torch.cuda.is_available() else "cpu"
 
 args = parser.parse_args()
 args.low_resource = True