coolkrishds commited on
Commit
f0d5b9d
·
1 Parent(s): 0392c67

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -1,7 +1,9 @@
1
  from transformers import pipeline
 
2
 
3
  # classifier = pipeline("sentiment-analysis")
4
  # res = classifier("I have been waiting for the transformer my whole life")
 
5
 
6
  generator = pipeline("text-generation", model="distilgpt2")
7
 
@@ -10,6 +12,5 @@ res = generator(
10
  max_length=30,
11
  num_return_sequences=2,
12
  )
13
- print(res)
14
-
15
 
 
1
  from transformers import pipeline
2
+ import streamlit as st
3
 
4
  # classifier = pipeline("sentiment-analysis")
5
  # res = classifier("I have been waiting for the transformer my whole life")
6
+ # print(res)
7
 
8
  generator = pipeline("text-generation", model="distilgpt2")
9
 
 
12
  max_length=30,
13
  num_return_sequences=2,
14
  )
15
+ st.text(res)
 
16