Testys commited on
Commit
f2f1ffa
1 Parent(s): 8678e61

Corrected issues

Browse files
Files changed (1) hide show
  1. main.py +2 -1
main.py CHANGED
@@ -99,7 +99,8 @@ async def generate_topic(
99
  img_file_object = app.state.url_utils.load_image(img_url)
100
 
101
  capt = app.state.img_caption.combo_model(img_file_object, text)
102
- return TopicResponse(topics=capt.topics, caption=capt.caption)
 
103
 
104
  raise HTTPException(
105
  status_code=status.HTTP_400_BAD_REQUEST,
 
99
  img_file_object = app.state.url_utils.load_image(img_url)
100
 
101
  capt = app.state.img_caption.combo_model(img_file_object, text)
102
+ print(capt)
103
+ return TopicResponse(topics=capt["topics"], caption=capt["caption"])
104
 
105
  raise HTTPException(
106
  status_code=status.HTTP_400_BAD_REQUEST,