Update app.py

app.py

@@ -17,7 +17,8 @@ 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: {os.listdir(model_dir)}', flush=True)
+model_dir = os.path.join(model_dir, 'model')
+
 tokenizer = AutoTokenizer.from_pretrained(model_dir, trust_remote_code=True, use_auth_token=True)
 model = AutoModelForCausalLM.from_pretrained(
     model_dir,