Shredder commited on
Commit
10fdf8a
1 Parent(s): f1bb772

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
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")