fix python 3.8 typing

bentoml · Jan 31, 2024 · e8f04e2 · e8f04e2
1 parent c6283a2
commit e8f04e2
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/service.py b/service.py
@@ -1,3 +1,7 @@
+from __future__ import annotations
+
+import typing as t
+
 import bentoml
 from PIL.Image import Image
 
@@ -19,7 +23,7 @@ def __init__(self) -> None:
         print("Model blip loaded", "device:", self.device)
 
     @bentoml.api
-    async def generate(self, img: Image, txt: str| None = None) -> str:
+    async def generate(self, img: Image, txt: t.Optional[str] = None) -> str:
         if txt:
             inputs = self.processor(img, txt, return_tensors="pt").to(self.device)
         else: