generation description out of picture

app.py

@@ -6,6 +6,7 @@ import torch
 st.set_page_config(layout="wide")
 
 image_pipe = pipeline("image-classification")
+text_pipe = pipeline("text-generation")
 k2t_tokenizer = AutoTokenizer.from_pretrained("mrm8488/t5-base-finetuned-common_gen")
 k2t_model = AutoModelWithLMHead.from_pretrained("mrm8488/t5-base-finetuned-common_gen")
 
@@ -32,6 +33,8 @@ with Image.open(img) as img:
 	keywords = ""
 	for keyword in results:
 		keywords += keyword["label"].split(',')[0]
+		
+	post_text = text_pipe(gen_sentence(keywords))[0]["generated_text"]
 	
 	col1, col2 = st.columns(2)
 	
@@ -41,4 +44,4 @@ with Image.open(img) as img:
 	
 	with col2:
 		st.subheader("Classification results")
-		st.write(gen_sentence(keywords))
+		st.write(post_text)