Create app.py

app.py

@@ -0,0 +1,54 @@
+# -*- coding: utf-8 -*-
+# file: deploy_demo.py
+# time: 2021/10/10
+# author: yangheng <yangheng@m.scnu.edu.cn>
+# github: https://github.com/yangheng95
+# Copyright (C) 2021. All Rights Reserved.
+
+import gradio as gr
+import pandas as pd
+
+from pyabsa import APCCheckpointManager
+
+sentiment_classifier = APCCheckpointManager.get_sentiment_classifier(checkpoint='multilingual',
+                                                                     auto_device=True  # False means load model on CPU
+                                                                     )
+
+
+def inference(text):
+    result = sentiment_classifier.infer(text=text,
+                                        print_result=True,
+                                        clear_input_samples=True)
+
+    result = pd.DataFrame({
+        'aspect': result[0]['aspect'],
+        'sentiment': result[0]['sentiment'],
+        'confidence': round(result[0]['confidence'], 3),
+        'ref_sentiment': result[0]['ref_sentiment'],
+        'is_correct': result[0]['ref_check'],
+    })
+
+    return result
+
+
+if __name__ == '__main__':
+    iface = gr.Interface(
+        fn=inference,
+        inputs=["text"],
+        examples=[
+            ['The [ASP]battery-life[ASP], and this [ASP]screen[ASP] is ok'],
+            ['The [ASP] battery-life [ASP] is bad'],
+            ['The [ASP] battery-life [ASP] is good'],
+            ['Strong build though which really adds to its [ASP]durability[ASP] .'],  # !sent! Positive
+            ['Strong [ASP]build[ASP] though which really adds to its durability . !sent! Positive'],
+            ['The [ASP]battery life[ASP] is excellent - 6-7 hours without charging . !sent! Positive'],
+            ['I have had my computer for 2 weeks already and it [ASP]works[ASP] perfectly . !sent! Positive'],
+            ['And I may be the only one but I am really liking [ASP]Windows 8[ASP] . !sent! Positive'],
+            ['This demo is trained on the laptop and restaurant and other review datasets from [ASP]ABSADatasets[ASP] (https://github.com/yangheng95/ABSADatasets)'],
+            ['To fit on your data, please train the model on your own data, see the [ASP]PyABSA[ASP] (https://github.com/yangheng95/PyABSA)'],
+        ],
+        outputs="dataframe",
+        title='Aspect Term Extraction for Short Texts (powered by PyABSA)'
+    )
+
+    iface.launch(share=True)