Spaces:

HYCCC
/

zero-shot_sentiment_analysis

Sleeping

cccc commited on Sep 12, 2023

Commit

cfea5fa

1 Parent(s): 1820058

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,7 +1,7 @@
 import gradio as gr
 from transformers import pipeline
-unmasker = pipeline('fill-mask', model='bert-base-uncased')
 def fill_mask(text):
   unmasked = unmasker(text)

 import gradio as gr
 from transformers import pipeline
+unmasker = pipeline('fill-mask', model="CCCC/ARCH_tuned_bert") #'bert-base-uncased')
 def fill_mask(text):
   unmasked = unmasker(text)