streetyogi commited on
Commit
82eb6ea
·
1 Parent(s): f494000

Update inference_server.py

Browse files
Files changed (1) hide show
  1. inference_server.py +1 -0
inference_server.py CHANGED
@@ -3,6 +3,7 @@ import logging
3
  import uvicorn
4
  from fastapi import FastAPI
5
  import transformers
 
6
 
7
  app = FastAPI()
8
 
 
3
  import uvicorn
4
  from fastapi import FastAPI
5
  import transformers
6
+ import torch
7
 
8
  app = FastAPI()
9