feat: add synopsis processing

app.py

@@ -11,7 +11,8 @@ def generate_synopsis(model, tokenizer, title):
     input_ids = tokenizer(title, return_tensors="tf")
     output = model.generate(input_ids['input_ids'], max_length=150, num_beams=5, no_repeat_ngram_size=2, top_k=50, attention_mask=input_ids['attention_mask'])
     synopsis = tokenizer.decode(output[0], skip_special_tokens=True)
-    return synopsis
+    processed_synopsis = "".join(synopsis.split(':')[1].rpartition('.')[:2]).strip()
+    return processed_synopsis
 
 favicon = "https://i.ibb.co/JRdhFZg/favicon-32x32.png"
 
@@ -28,6 +29,6 @@ button_synopsis = st.button('Get synopsis')
 if button_synopsis:
     if len(prod_title.split(' ')) > 0:
         gen_synopsis = generate_synopsis(model, tokenizer, f"{prod_title} : ")
-        st.text_area(gen_synopsis, disabled=True)
+        st.text_area('Generated synopsis', value=gen_synopsis, disabled=True)
     else:
         st.write('Write a title for the generator to work !')