Aviral Jain
commited on
Update api.py
Browse files
api.py
CHANGED
|
@@ -9,12 +9,14 @@ from transformers import AutoTokenizer, AutoModelForSequenceClassification
|
|
| 9 |
from pytorch_forecasting import TemporalFusionTransformer
|
| 10 |
from bs4 import BeautifulSoup
|
| 11 |
import requests
|
| 12 |
-
import torch
|
| 13 |
from dotenv import load_dotenv
|
| 14 |
import os
|
| 15 |
from fastapi.middleware.cors import CORSMiddleware
|
| 16 |
|
| 17 |
os.environ["CUDA_VISIBLE_DEVICES"] = "-1"
|
|
|
|
|
|
|
|
|
|
| 18 |
device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
|
| 19 |
|
| 20 |
MODEL_PATH = "lib/20_lstm_model.h5"
|
|
|
|
| 9 |
from pytorch_forecasting import TemporalFusionTransformer
|
| 10 |
from bs4 import BeautifulSoup
|
| 11 |
import requests
|
|
|
|
| 12 |
from dotenv import load_dotenv
|
| 13 |
import os
|
| 14 |
from fastapi.middleware.cors import CORSMiddleware
|
| 15 |
|
| 16 |
os.environ["CUDA_VISIBLE_DEVICES"] = "-1"
|
| 17 |
+
|
| 18 |
+
import torch
|
| 19 |
+
|
| 20 |
device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
|
| 21 |
|
| 22 |
MODEL_PATH = "lib/20_lstm_model.h5"
|