roman commited on
Commit
06c4ac4
·
1 Parent(s): 6e39d3b

add transformer example

Browse files
Files changed (1) hide show
  1. app.py +26 -4
app.py CHANGED
@@ -1,8 +1,30 @@
 
 
 
 
 
 
1
  import streamlit as st
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
2
 
3
- x = st.slider('Select a value')
4
- st.write(x, 'squared is', x * x)
5
 
6
 
7
- if __name__ == '__main__':
8
- st.title('ACR')
 
1
+ # import streamlit as st
2
+ #
3
+ # x = st.slider('Select a value')
4
+ # st.write(x, 'squared is', x * x)
5
+
6
+
7
  import streamlit as st
8
+ from transformers import pipeline
9
+
10
+ # Load a pre-trained pipeline for text generation
11
+ generator = pipeline('text-generation', model='gpt2')
12
+
13
+ st.title('Simple Hugging Face Space with Streamlit')
14
+
15
+ # Text input widget
16
+ input_text = st.text_area('Enter your text here:', '')
17
+
18
+ if st.button('Generate'):
19
+ if input_text:
20
+ # Generate text using the model
21
+ outputs = generator(input_text, max_length=50, num_return_sequences=1)
22
+ st.write('Generated Text:')
23
+ st.write(outputs[0]['generated_text'])
24
+ else:
25
+ st.write('Please enter some text to generate.')
26
 
27
+ # if __name__ == '__main__':
28
+ # st.title('ACR')
29
 
30