Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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
|
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)
|