indoflaven commited on
Commit
40d2ebc
·
1 Parent(s): ad4e573

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -24,7 +24,7 @@ def read_root():
24
  def return_item(item: Item):
25
  return item
26
 
27
- @app.put("/")
28
  def predict(item: Item):
29
  # Load the OpenAI CLIP Model
30
  print('Loading CLIP Model...')
 
24
  def return_item(item: Item):
25
  return item
26
 
27
+ @app.put("/", response_model=ScoreResponse)
28
  def predict(item: Item):
29
  # Load the OpenAI CLIP Model
30
  print('Loading CLIP Model...')