Spaces:

kushan98
/

comment_review

Runtime error

kushan98 commited on May 10, 2023

Commit

8b19034

•

1 Parent(s): 8a5c7b2

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -2,8 +2,8 @@ import streamlit as st
 from transformers import pipeline
 def summarize_text(text):
-    unmasker = pipeline('fill-mask', model='distilbert-base-uncased')
-    text_input = unmasker(text, do_sample=True)[0]['text_input']
     return text_input
 def main():

 from transformers import pipeline
 def summarize_text(text):
+    unmasker = pipeline('fill-mask', model='casarf/comment_model_test')
+    text_input = unmasker(text)
     return text_input
 def main():