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) | |