Srisamorn commited on
Commit
1266462
·
1 Parent(s): 0a332cc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -2,9 +2,9 @@ import streamlit as st
2
  from transformers import pipeline
3
  from datasets import load_dataset
4
 
5
- classifier = pipeline("text-generation", model=AutoModelForCausalLM.from_pretrained("xlnet-base-cased"))
6
  def main():
7
- st.title("Text review")
8
  with st.form("text_field"):
9
  text = st.text_area('enter some text:')
10
  # clicked==True only when the button is clicked
 
2
  from transformers import pipeline
3
  from datasets import load_dataset
4
 
5
+ classifier = pipeline("text-generation", model="gpt2-medium")
6
  def main():
7
+ st.title("Text generation")
8
  with st.form("text_field"):
9
  text = st.text_area('enter some text:')
10
  # clicked==True only when the button is clicked