cracker0935 commited on
Commit
bd941f5
·
verified ·
1 Parent(s): 9c67488

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +2 -1
main.py CHANGED
@@ -7,7 +7,8 @@ from transformers import AutoTokenizer
7
  from typing import List
8
  from fastapi.middleware.cors import CORSMiddleware
9
  import os
10
- import requests # <--- Added this
 
11
 
12
  from model_arch import ResearchHybridModel
13
  from preprocessing import ChaParser
 
7
  from typing import List
8
  from fastapi.middleware.cors import CORSMiddleware
9
  import os
10
+ import requests
11
+ import tempfile
12
 
13
  from model_arch import ResearchHybridModel
14
  from preprocessing import ChaParser