lainlives commited on
Commit
91ff0d2
·
verified ·
1 Parent(s): 6cab965

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -2,6 +2,7 @@ from fastapi import FastAPI, Depends, HTTPException, status
2
  from fastapi.security import HTTPBearer, HTTPAuthorizationCredentials
3
  from pydantic import BaseModel
4
  from typing import List, Optional
 
5
  import torch
6
  from transformers import AutoModelForCausalLM, AutoTokenizer, BitsAndBytesConfig
7
  HF_TOKEN = os.getenv('HF_TOKEN')
@@ -64,4 +65,4 @@ async def chat_endpoint(request: ChatRequest, token: str = Depends(validate_api_
64
 
65
  if __name__ == "__main__":
66
  import uvicorn
67
- uvicorn.run(app, host="0.0.0.0", port=8000)
 
2
  from fastapi.security import HTTPBearer, HTTPAuthorizationCredentials
3
  from pydantic import BaseModel
4
  from typing import List, Optional
5
+ import os
6
  import torch
7
  from transformers import AutoModelForCausalLM, AutoTokenizer, BitsAndBytesConfig
8
  HF_TOKEN = os.getenv('HF_TOKEN')
 
65
 
66
  if __name__ == "__main__":
67
  import uvicorn
68
+ uvicorn.run(app, host="0.0.0.0", port=7860)