Update app.py
Browse files
app.py
CHANGED
@@ -2,7 +2,7 @@ import gradio as gr
|
|
2 |
from transformers import AutoTokenizer, AutoModelForTokenClassification,RobertaTokenizer
|
3 |
import torch
|
4 |
from fin_readability_sustainability import BERTClass, do_predict
|
5 |
-
|
6 |
|
7 |
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
8 |
|
@@ -29,8 +29,8 @@ def get_sustainability(text):
|
|
29 |
|
30 |
|
31 |
|
32 |
-
b6 = gr.Button("Get Sustainability")
|
33 |
-
b6.click(get_sustainability, inputs = text, outputs = gr.HighlightedText())
|
34 |
|
35 |
|
36 |
iface = gr.Interface(fn=get_sustainability, inputs=gr.inputs.Textbox(lines=5, placeholder="Enter Financial Text here..."), title="CONBERT",description="SUSTAINABILITY TOOL", outputs=gr.HighlightedText(), allow_flagging="never")
|
|
|
2 |
from transformers import AutoTokenizer, AutoModelForTokenClassification,RobertaTokenizer
|
3 |
import torch
|
4 |
from fin_readability_sustainability import BERTClass, do_predict
|
5 |
+
import pandas as pd
|
6 |
|
7 |
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
8 |
|
|
|
29 |
|
30 |
|
31 |
|
32 |
+
# b6 = gr.Button("Get Sustainability")
|
33 |
+
#b6.click(get_sustainability, inputs = text, outputs = gr.HighlightedText())
|
34 |
|
35 |
|
36 |
iface = gr.Interface(fn=get_sustainability, inputs=gr.inputs.Textbox(lines=5, placeholder="Enter Financial Text here..."), title="CONBERT",description="SUSTAINABILITY TOOL", outputs=gr.HighlightedText(), allow_flagging="never")
|