| import streamlit as st | |
| from transformers import pipeline | |
| poet = pipeline('text-generation', | |
| model='ghosh-r/robi-kobi', | |
| tokenizer='ghosh-r/bangla-gpt2') | |
| poem = poet('দরজা ধরে দাঁড়াও তুমি', max_length=250) | |
| output = poem[0]['generated_text'].replace('\n', ' ').replace('\xa0', '') | |
| st.write(output) | |