Spaces:
Sleeping
Sleeping
Commit
·
96eea2a
1
Parent(s):
54d4ac0
Small efficiency changes on the classification returns and extra info
Browse filesNo need to do two loops which do basically the same thing to `classifications_input` and `classifications_summary`
app.py
CHANGED
|
@@ -20,12 +20,9 @@ def do_the_thing(input, hypothesis, labels):
|
|
| 20 |
hypothesis= "This example is {}."
|
| 21 |
summarisation = summary(input, truncation=True)[0]['summary_text']
|
| 22 |
zsc_results = oracle(sequences=[input, summarisation], candidate_labels=labels, multi_label=False, batch_size=2, hypothesis_template=hypothesis)
|
| 23 |
-
classifications_input = {}
|
| 24 |
for i in range(len(labels)):
|
| 25 |
classifications_input.update({zsc_results[0]['labels'][i]: zsc_results[0]['scores'][i]})
|
| 26 |
-
i+=1
|
| 27 |
-
classifications_summary = {}
|
| 28 |
-
for i in range(len(labels)):
|
| 29 |
classifications_summary.update({zsc_results[1]['labels'][i]: zsc_results[1]['scores'][i]})
|
| 30 |
i+=1
|
| 31 |
return [summarisation, classifications_input, classifications_summary]
|
|
@@ -36,7 +33,7 @@ with gr.Blocks() as frontend:
|
|
| 36 |
btn_submit = gr.Button(value="Summarise and Classify")
|
| 37 |
with gr.Row():
|
| 38 |
with gr.Column():
|
| 39 |
-
input_labels = gr.Dropdown(label="Classification Labels", choices=labels, multiselect=True, value=selected_labels, interactive=True, allow_custom_value=True, info="Labels to classify the original text and summary")
|
| 40 |
input_hypothesis = gr.Textbox(label="Hypothesis Template", info="This must include the {} format syntax. Blank and invalid inputs get defaulted to the palceholder text.", value="This git commit relates to {} changes.", placeholder="This example is {}.")
|
| 41 |
with gr.Column():
|
| 42 |
output_summary_text = gr.TextArea(label="Summary of Notes")
|
|
|
|
| 20 |
hypothesis= "This example is {}."
|
| 21 |
summarisation = summary(input, truncation=True)[0]['summary_text']
|
| 22 |
zsc_results = oracle(sequences=[input, summarisation], candidate_labels=labels, multi_label=False, batch_size=2, hypothesis_template=hypothesis)
|
| 23 |
+
classifications_summary, classifications_input = {}, {}
|
| 24 |
for i in range(len(labels)):
|
| 25 |
classifications_input.update({zsc_results[0]['labels'][i]: zsc_results[0]['scores'][i]})
|
|
|
|
|
|
|
|
|
|
| 26 |
classifications_summary.update({zsc_results[1]['labels'][i]: zsc_results[1]['scores'][i]})
|
| 27 |
i+=1
|
| 28 |
return [summarisation, classifications_input, classifications_summary]
|
|
|
|
| 33 |
btn_submit = gr.Button(value="Summarise and Classify")
|
| 34 |
with gr.Row():
|
| 35 |
with gr.Column():
|
| 36 |
+
input_labels = gr.Dropdown(label="Classification Labels", choices=labels, multiselect=True, value=selected_labels, interactive=True, allow_custom_value=True, info="Labels to classify the original text and summary. Select more or add your own.")
|
| 37 |
input_hypothesis = gr.Textbox(label="Hypothesis Template", info="This must include the {} format syntax. Blank and invalid inputs get defaulted to the palceholder text.", value="This git commit relates to {} changes.", placeholder="This example is {}.")
|
| 38 |
with gr.Column():
|
| 39 |
output_summary_text = gr.TextArea(label="Summary of Notes")
|