dbmoradi60 commited on
Commit
1cf2a4d
·
verified ·
1 Parent(s): 10bd6ea

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -64,7 +64,7 @@ except Exception as e:
64
  print("Loading tokenizer...")
65
  try:
66
  tokenizer = AutoTokenizer.from_pretrained(
67
- MODEL_ID, # Load directly from Hub to avoid local config issues
68
  cache_dir=cache_dir,
69
  trust_remote_code=True
70
  )
@@ -133,4 +133,4 @@ torch.cuda.empty_cache()
133
 
134
  if __name__ == "__main__":
135
  import uvicorn
136
- uvicorn.run(app, host="0.0.0.0", port=8007)
 
64
  print("Loading tokenizer...")
65
  try:
66
  tokenizer = AutoTokenizer.from_pretrained(
67
+ MODEL_ID, # Load directly from Hub
68
  cache_dir=cache_dir,
69
  trust_remote_code=True
70
  )
 
133
 
134
  if __name__ == "__main__":
135
  import uvicorn
136
+ uvicorn.run(app, host="0.0.0.0", port=8000)