Spaces:

kina006097
/

kids-playground-ai-api

Sleeping

kina006097 commited on Aug 30

Commit

5a8f8f9

1 Parent(s): f079f59

ruffチェック

Files changed (1) hide show

src/ai_api/core/inference.py CHANGED Viewed

@@ -30,4 +30,4 @@ class Summarizer:
             input_ids, max_length=50, min_length=10, do_sample=False
         )
         summary = self.tokenizer.decode(output_ids[0], skip_special_tokens=True)
-        return cast(str, summary)

             input_ids, max_length=50, min_length=10, do_sample=False
         )
         summary = self.tokenizer.decode(output_ids[0], skip_special_tokens=True)
+        return cast(str, summary)