Ashrafb commited on
Commit
17c8950
·
verified ·
1 Parent(s): ac21aae

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -12,7 +12,7 @@ app = FastAPI()
12
 
13
 
14
 
15
- model = cacartoon1(device='cuda' if torch.cuda.is_available() else 'cpu')
16
 
17
  def generate_cartoon(image_bytes: bytes) -> bytes:
18
  image = Image.open(BytesIO(image_bytes))
 
12
 
13
 
14
 
15
+ model = cartoon1(device='cuda' if torch.cuda.is_available() else 'cpu')
16
 
17
  def generate_cartoon(image_bytes: bytes) -> bytes:
18
  image = Image.open(BytesIO(image_bytes))