Ritobrata Ghosh commited on
Commit
c43295a
1 Parent(s): a0357c9

first test

Browse files
Files changed (1) hide show
  1. app.py +12 -2
app.py CHANGED
@@ -1,4 +1,14 @@
1
  import streamlit as st
2
 
3
- x = st.slider('Select a value')
4
- st.write(x, ' squared is ', x * x)
 
 
 
 
 
 
 
 
 
 
1
  import streamlit as st
2
 
3
+ from transformers import pipeline
4
+
5
+ poet = pipeline('text-generation',
6
+ model='ghosh-r/robi-kobi',
7
+ tokenizer='ghosh-r/bangla-gpt2')
8
+
9
+ poem = poet('দরজা ধরে দাঁড়াও তুমি', max_length=250)
10
+
11
+ output = poem[0]['generated_text'].replace('\n', ' ').replace('\xa0', '')
12
+
13
+
14
+ st.write(output)