Update main.py
Browse files
main.py
CHANGED
|
@@ -5,6 +5,12 @@ import torch
|
|
| 5 |
from fastapi.middleware.cors import CORSMiddleware
|
| 6 |
|
| 7 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 8 |
# Define the FastAPI app
|
| 9 |
app = FastAPI()
|
| 10 |
|
|
|
|
| 5 |
from fastapi.middleware.cors import CORSMiddleware
|
| 6 |
|
| 7 |
|
| 8 |
+
import os
|
| 9 |
+
access_token_read = os.getenv('DS4')
|
| 10 |
+
|
| 11 |
+
from huggingface_hub import login
|
| 12 |
+
login(token = access_token_read)
|
| 13 |
+
|
| 14 |
# Define the FastAPI app
|
| 15 |
app = FastAPI()
|
| 16 |
|