Update app.py

app.py

@@ -100,7 +100,12 @@ def fls(text):
 ner=pipeline('ner',model='Jean-Baptiste/camembert-ner-with-dates',tokenizer='Jean-Baptiste/camembert-ner-with-dates', aggregation_strategy="simple")
 def fin_ner(text):
     replaced_spans = ner(text)
-    return {"text": text, "entities": replaced_spans}
+    new_spans=[]
+    for item in replaced_spans:
+        item['entity']=item['entity_group']
+        del item['entity_group']
+        new_spans.append(item)
+    return {"text": text, "entities": new_spans}
     
      
 #CUAD STARTS    
@@ -116,7 +121,8 @@ def load_questions_short():
     with open('questionshort.txt') as f:
         questions_short = f.readlines()
     return questions_short
-
+questions = load_questions()
+questions_short = load_questions_short()
 def quad(query,file):
     with open(file.name) as f:
         paragraph = f.read()
@@ -150,7 +156,7 @@ def quad(query,file):
 #iface = gr.Interface(fn=get_sustainability, inputs="textbox", title="CONBERT",description="SUSTAINABILITY TOOL", outputs=gr.HighlightedText(), allow_flagging="never")
 #iface.launch()
 
-iface = gr.Interface(fn=quad, inputs=[gr.inputs.Textbox(label='SEARCH QUERY'),gr.inputs.File(label='TXT FILE')], title="CONBERT",description="SUSTAINABILITY TOOL",article='Article', outputs=[gr.outputs.Textbox(label='Answer'),gr.outputs.Textbox(label='Summary'),gr.HighlightedText(label='NER'),gr.HighlightedText(label='CLAIM'),gr.HighlightedText(label='SUSTAINABILITY'),gr.HighlightedText(label='FLS')], allow_flagging="never")
+iface = gr.Interface(fn=quad, inputs=[gr.Dropdown(choices=questions,label='SEARCH QUERY'),gr.inputs.File(label='TXT FILE')], title="CONBERT",description="SUSTAINABILITY TOOL",article='Article', outputs=[gr.outputs.Textbox(label='Answer'),gr.outputs.Textbox(label='Summary'),gr.HighlightedText(label='NER'),gr.HighlightedText(label='CLAIM'),gr.HighlightedText(label='SUSTAINABILITY'),gr.HighlightedText(label='FLS')], allow_flagging="never")
 
 
 iface.launch()