valhalla commited on
Commit
16e3302
1 Parent(s): 11b197a

Update server.py

Browse files
Files changed (1) hide show
  1. server.py +1 -0
server.py CHANGED
@@ -4,6 +4,7 @@ import base64
4
  from io import BytesIO
5
  sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
6
 
 
7
  from fastapi import FastAPI
8
  import numpy as np
9
  from PIL import Image
 
4
  from io import BytesIO
5
  sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
6
 
7
+ import torch
8
  from fastapi import FastAPI
9
  import numpy as np
10
  from PIL import Image