Update app.py
Browse files
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=
|
|
|
|
| 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)
|