akhaliq HF Staff commited on
Commit
a11cb66
·
1 Parent(s): 288b222

feat: add return type annotation to predict function in app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -35,7 +35,7 @@ app = Server()
35
 
36
  @app.api()
37
  @spaces.GPU(duration=120)
38
- def predict(message: str, file: FileData = None, downsample_mode: str = "16x"):
39
  """
40
  General inference endpoint for both image and video.
41
  """
 
35
 
36
  @app.api()
37
  @spaces.GPU(duration=120)
38
+ def predict(message: str, file: FileData = None, downsample_mode: str = "16x") -> str:
39
  """
40
  General inference endpoint for both image and video.
41
  """