Update app.py

app.py

@@ -4,8 +4,27 @@ model_id = "knkarthick/TOPIC-DIALOGSUM"
 
 generator = pipeline(task="text2text-generation", model=model_id)
 
+def split_paragraph(paragraph, max_chunk_size=1024):
+    words = paragraph.split()
+    chunks = []
+    current_chunk = []
+
+    for word in words:
+        if len(current_chunk) + len(word) + 1 <= max_chunk_size:
+            current_chunk.append(word)
+        else:
+            chunks.append(' '.join(current_chunk))
+            current_chunk = [word]
+
+    if current_chunk:
+        chunks.append(' '.join(current_chunk))
+
+    return chunks
+
 def launch(input):
-    return generator(input)
+    if len(input) > 1024:
+        return " ".join([res.generated_text for res in generator(split_paragraph(input))])
+    return generator(input)[0].generated_text
 
 iface = gr.Interface(launch, inputs="text", outputs="text")
 iface.launch()