Spaces:

Waraporn
/

text-generation

Runtime error

Waraporn commited on Oct 5, 2022

Commit

ddaf92b

1 Parent(s): 623724b

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,9 +1,9 @@
 import streamlit as st
 from transformers import pipeline
-classifier = pipeline("sentiment-analysis", model="Donlapark/finetuned_yelp")
 def main():
-    st.title("Yelp review")
     with st.form("text_field"):
         text = st.text_area('enter some text:')

 import streamlit as st
 from transformers import pipeline
+classifier = pipeline("text-generation", model = joblib.load('model.joblib'))
 def main():
+    st.title("Text generation")
     with st.form("text_field"):
         text = st.text_area('enter some text:')