Aviral Jain commited on
Commit
d2c1dba
·
verified ·
1 Parent(s): 7add2a4

Update api.py

Browse files
Files changed (1) hide show
  1. api.py +3 -1
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"