Update app.py

app.py

@@ -17,7 +17,7 @@ device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 huggingface_hub.login(token=HF_TOKEN)
 
 model_dir = huggingface_hub.snapshot_download(repo_id=REPO_ID, allow_patterns=["model/*"])
-print(f'AOBA: {model_dir}', flush=True)
+print(f'AOBA: {os.listdir(model_dir)}', flush=True)
 tokenizer = AutoTokenizer.from_pretrained(model_dir, trust_remote_code=True, use_auth_token=True)
 model = AutoModelForCausalLM.from_pretrained(
     model_dir,