Fix with https://github.com/jeanigarcia/ernie_demo_local/blob/master/app.py

app.py

@@ -1,11 +1,19 @@
 import gradio as gr
-import tensorflow
-import transformers
-import sklearn
+from ernie.ernie import SentenceClassifier
+from ernie import helper
 
+file_list=['config.json', 'tf_model.h5', 'tokenizer.json', 'vocab.txt', 'special_tokens_map.json', 'tokenizer_config.json']
 
-def greet(name):
-    return "Hello " + name + "!!"
+for f in file_list:
+    helper.download_from_hub(repo_id='jeang/bert-finetuned-sentence-classification-toy', filename=f, cache_dir='model/')
+    
+classifier = SentenceClassifier(model_path='model/', max_length=128, labels_no=2)
 
-iface = gr.Interface(fn=greet, inputs="text", outputs="text")
-iface.launch()
+def classify(sentence):
+    probability = classifier.predict_one(sentence)[1]
+    
+    return 'probability = ' + str(probability) + ' (' + ('positive' if probability >= 0.5 else 'negative') +  ')'
+            
+
+iface = gr.Interface(fn=classify, inputs="text", outputs="text")
+iface.launch()