sagar008 commited on
Commit
9da1804
·
verified ·
1 Parent(s): f2fe324

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -3
app.py CHANGED
@@ -6,10 +6,8 @@ import uvicorn
6
 
7
  app = FastAPI()
8
 
9
- # Read Hugging Face token from environment
10
  HF_AUTH_TOKEN = os.getenv("HF_TOKEN")
11
 
12
- # Load model and tokenizer using auth token
13
  MODEL_NAME = "VincentMuriuki/legal-summarizer"
14
  summarizer = pipeline("summarization", model=MODEL_NAME, use_auth_token=HF_AUTH_TOKEN)
15
  tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME, use_auth_token=HF_AUTH_TOKEN)
@@ -38,5 +36,5 @@ def chunk_text(data: ChunkInput):
38
 
39
  return {"chunks": chunks}
40
 
41
- if _name_ == "_main_":
42
  uvicorn.run(app, host="0.0.0.0", port=7860)
 
6
 
7
  app = FastAPI()
8
 
 
9
  HF_AUTH_TOKEN = os.getenv("HF_TOKEN")
10
 
 
11
  MODEL_NAME = "VincentMuriuki/legal-summarizer"
12
  summarizer = pipeline("summarization", model=MODEL_NAME, use_auth_token=HF_AUTH_TOKEN)
13
  tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME, use_auth_token=HF_AUTH_TOKEN)
 
36
 
37
  return {"chunks": chunks}
38
 
39
+ if __name__ == "__main__":
40
  uvicorn.run(app, host="0.0.0.0", port=7860)