File size: 351 Bytes
a0357c9 c43295a |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 |
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)
|