g3casey commited on
Commit
d662828
·
1 Parent(s): fb4f8c6

change to get_wiki of summarize

Browse files
Files changed (1) hide show
  1. app.py +8 -4
app.py CHANGED
@@ -44,8 +44,12 @@ def get_wiki(search_term):
44
  # model = AutoModelForSeq2SeqLM.from_pretrained("sgugger/my-awesome-model")
45
 
46
 
47
-
48
-
49
-
50
- iface = gr.Interface(fn=get_wiki, inputs="text", outputs=["text",'orig_length','sum_length'])
 
 
 
 
51
  iface.launch()
 
44
  # model = AutoModelForSeq2SeqLM.from_pretrained("sgugger/my-awesome-model")
45
 
46
 
47
+ out_sum_text = gr.Textbox(label='Summarized Text')
48
+ out_orig_test_len = gr.Number(label='Original Text Length')
49
+ out_sum_text_len = gr.Number(label='Summarized Text Length')
50
+
51
+ iface = gr.Interface(fn=get_wiki,
52
+ inputs=gr.Textbox(lines=2, placeholder="Wikipedia search term here..."),
53
+ outputs=[out_sum_text,out_orig_test_len,out_sum_text_len]
54
+ )
55
  iface.launch()