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")
|