Update app.py

app.py

@@ -548,13 +548,20 @@ def upscaler(
 
     return enhanced_image
 
+def get_tensor_length(tensor):
+    nums = list(tensor.size)
+    ret = 1
+    for num in nums:
+        ret = ret * num
+    return ret
+
 def summarize_text(
     text, max_len=20, min_len=10
 ):
     log(f'CALL summarize_text')
-    inputs = tokenizer("summarize: " + text, return_tensors="pt", max_length=float('inf'), truncation=False)
+    inputs = tokenizer.encode("summarize: " + text, return_tensors="pt", max_length=float('inf'), truncation=False)
     i = 1
-    while len(inputs[0]) > max_len:
+    while get_tensor_length(inputs) > max_len:
         print(f'DBG summarize_text 1 {i}')
         outputs = model.generate(
             inputs[:512],
@@ -562,7 +569,7 @@ def summarize_text(
             num_beams=4,
             early_stopping=True
         )
-        inputs[0] = outputs[0] + inputs[0][512:]
+        inputs[0] = [*list(outputs[0]), *list(inputs[0][512:])]
         i = i + 1
     summary = tokenizer.decode(inputs[0])
     log(f'RET summarize_text with summary as {summary}')