enrique commited on
Commit
d389798
1 Parent(s): b903bde

Adding transformers pipeline example

Browse files
Files changed (2) hide show
  1. app.py +7 -2
  2. requirements.txt +2 -0
app.py CHANGED
@@ -1,4 +1,9 @@
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
+ from transformers import pipeline
3
 
4
+ pipe = pipeline("summarization")
5
+ text = st.text_area("enter some text")
6
+
7
+ if text:
8
+ out = pipe(text)
9
+ st.json(out)
requirements.txt ADDED
@@ -0,0 +1,2 @@
 
 
 
1
+ torch
2
+ transformers